win10 安装 tensorflow-gpu 2.10.0_conda安装tensorflow2.10.0 gpu

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

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

安装 conda

安装cuda

安装python3.10

conda create -p D:\tensorEnv python=3.10
1

激活虚拟环境

conda activate  D:\tensorEnv
1

安装cuda

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

conda install cudatoolkit=12.2
1

安装cudnn

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

conda install cudnn=8.9
1

安装tensorflow的GPU版本

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

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

测试 test-gpu.py

import tensorflow as tf

# print(tf.test.is_gpu_available())
print(tf.config.list_physical_devices('GPU'))
1
2
3
4

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

输出

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

参考：
适用于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