安装CUDA, cuDnn, Tensorflow-gpu
1. CUDA cuDnn, Tensorflow-gpu 版本要对应
Tensorflow官网给出的对应关系
一定要注意版本对应关系
比如我要安装Tensorflow-gpu 2.3.0,对应Python版本3.5~3.8, cuDnn版本7.6, CUDA版本10.1
CUDA下载地址
cuDnn下载地址
pip安装Tensorflow-gpu:
pip3 install tensorflow-gpu==2.3.0
2. 环境变量设置
Tensorflow官网给出的环境变量设置规则
CUDA和cuDn
原创
2020-11-03 18:10:17 ·
240 阅读 ·
0 评论