win10 安装 tensorflow-gpu 2.10.0-Toy模板网

这篇具有很好参考价值的文章主要介绍了win10 安装 tensorflow-gpu 2.10.0。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

win10 安装 tensorflow-gpu 2.10.0

系统配置

系统 win10 x64
显卡 GTX 1660 Ti
CUDA 12.2
cudnn 8.9

查看版本对应：

https://tensorflow.google.cn/install/source_windows#gpu

Version	Python version	Compiler	Build tools	cuDNN	CUDA
tensorflow_gpu-2.10.0	3.7-3.10	MSVC 2019	Bazel 5.1.1	8.1	11.2

这里查看我系统配置 我安装 python 3.10 和 tensorflow_gpu-2.10.0

安装 conda

安装cuda

安装python3.10

conda create -p D:\tensorEnv python=3.10

激活虚拟环境

conda activate  D:\tensorEnv

安装cuda

这里根据自己的版本安装，按照表格推荐 11.2 ，我这里是12.2

conda install cudatoolkit=12.2

安装cudnn

这里根据自己的版本安装，按照表格推荐 8.1 ，我这里是 8.9

conda install cudnn=8.9

安装tensorflow的GPU版本

这里根据自己的版本安装，按照表格推荐tensorflow-gpu==2.10.0

(D:\tensorEnv)  C:\Users\mingxingwang> python -m pip install tensorflow-gpu==2.10.0

测试 test-gpu.py

import tensorflow as tf

# print(tf.test.is_gpu_available())
print(tf.config.list_physical_devices('GPU'))

(D:\tensorEnv)  C:\Users\mingxingwang> python  test-gpu.py

输出

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

参考：
适用于Microsoft窗口的 CUDA 安装指南 https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/

cuDNN Archive https://developer.nvidia.com/rdp/cudnn-archive

Build a TensorFlow pip package from source and install it on Windows https://tensorflow.google.cn/install/source_windows#gpu文章来源地址https://www.toymoban.com/news/detail-631487.html

到了这里，关于win10 安装 tensorflow-gpu 2.10.0的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！

win10 安装 tensorflow-gpu 2.10.0