首先选择tensorflow对应的cuda和cudnn
但是CUDA11.2.X不支持win11,CUDA11.8.X版本却可以兼容w11。
那我们标准为Tensorflow2.10.0+cuda11.8+cudnn8.8
欧克,选择好了,开干。
创建python3.10
conda create -n tf210 python==3.10.0
激活虚拟环境
conda activate tf210
添加中科大镜像源
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
这些命令将显示当前使用的镜像源,并将通道优先级设置为严格,确保首先从中科大的镜像源获取软件包。
conda config --set show_channel_urls yes
conda config --set channel_priority strict
添加完后下载cuda11.8+cudnn8.8,下载不成功多换几个镜像源
conda install cudatoolkit=11.8 cudnn=8.8
两个都下好之后需要添加环境变量,在系统变量里的path添加
C:\ProgramData\Anaconda3\envs\name\Library\bin
name是你的虚拟环境名字
下载tensorflow2.10.0
pip install tensorflow-gpu==2.10 -i https://pypi.tuna.tsinghua.edu.cn/simple/
下完了测试一下
python
import tensorflow as tf
print(tf.__version__)
gpus =tf.config.list_physical_devices('GPU')
print(gpus)
战斗,爽!
踩坑事项:
1.注意,虚拟环境用python是可以识别到gpu,但是到了jupyter里就识别不到了,后来才知道需要添加环境变量就ok了
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
2.我本地C盘是有cuda12.3的所以不用担心会不会冲突。
3.最开始我更新cuda12.3的时候踩了很多坑,比如cuda四大杀手。
我的经验是把全部的删掉
再重新下载,因为这样才很简单而且还不容易报错
4.Windows Installer 错误。
无法访问.Windows Installer服务
解决 可恶的联想管家启动项没开
哈哈哈哈哈
参考:
还在系统中安装CUDA?Anaconda虚拟环境中安装CUDA配合Tensorflow-gpu-多版本 - 简书 (jianshu.com)
Tensorflow2.X+cuda+cudnn配置指南(RTX4060+win11+Anaconda3) - MistLuoy - 博客园 (cnblogs.com)