我的工具版本分别是: CUDA9.0 cudnn7.4.1.5 tensorflow 1.12.0 vs2015(update3) python 3.6,严格按照此版本配置可以确保安装成功
其它版本的CUDA和cudnn 测试未能成功,尤其是cudnn的版本难以把握用哪个。
如下面这个配置可以肯定是不成功的
- tensorflow 1.12.0
- cuda toolkit 10.0.130
- cudnn 7.4.1.5
下面这个高版本的网友测试是成功的
tensorflow-gpu==2.0.0-alpha0
python 3.6.8
CUDA v10.0
cudnn 7.4.1.5
1 使用命令 pip install tensorflow-gpu==1.12.0 等待安装成功
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include;%PATH%
SET PATH=C:\tools\cuda\bin;%PATH%
一些网友测试的成功和失败的例子如下所示
FAIL
tensorflow-gpu==r1.13
python 3.7.3
CUDA v10.1
cudnn 7.5.0.56
FAIL
tensorflow-gpu==2.0.0-alpha0
python 3.7.3
CUDA v10.1
cudnn 7.5.0.56
FAIL
tensorflow-gpu==2.0.0-alpha0
python 3.6.8
CUDA v10.1
cudnn 7.5.0.56
SUCCESS
tensorflow-gpu==2.0.0-alpha0
python 3.6.8
CUDA v10.0
cudnn 7.4.1.5