1. CUDA cuDnn, Tensorflow-gpu 版本要对应
Tensorflow官网给出的对应关系
一定要注意版本对应关系
比如我要安装Tensorflow-gpu 2.3.0
,对应Python
版本3.5~3.8, cuDnn
版本7.6, CUDA
版本10.1
pip
安装Tensorflow-gpu
:
pip3 install tensorflow-gpu==2.3.0
2. 环境变量设置
Tensorflow官网给出的环境变量设置规则
CUDA
和cuDnn
安装完成后,计算机图标上点右键,打开属性->高级系统设置->环境变量,系统中已经有了两条:
CUDA_PATH=E:\NVIDIA Computing Toolkit\CUDA\v10.1
CUDA_PATH_V10_1=E:\NVIDIA Computing Toolkit\CUDA\v10.1
我们要在系统变量path
里添加CUDA
:
PATH=%你自己安装的路径%\NVIDIA Computing Toolkit\CUDA\v10.1\bin
PATH=%你自己安装的路径%\NVIDIA Computing Toolkit\CUDA\v10.1\libnvvp
再添加cuDnn
变量:
PATH=%你自己解压的路径%\cuda\bin