windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0

安装 Visual Studio 2019下载地址
下载完成后点击exe,勾选通用Windows平台开发和使用c++的桌面开发,直到安装完成。
windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第1张图片
右键点击NVIDIA控制面板,进入帮助->系统信息,选择组件,查看NVCUDA.dll上写的CUDA版本号
windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第2张图片
我的Nvdia控制面板上写的是CUDA10.1

我安装的是CUDA10.1和对应的Cudnn7.6.4(查看CUDA和Cudnn对应的版本,需要登陆一下)
CUDA默认安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

以上完成之后把以下四个路径加入到环境变量中:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib

安装完成后,在C盘的用户目录下,选择你当前的用户进去新建文件夹pip,再在里面建立文件pip.ini,在里面设置镜像源和信任,如下

[global] 
index-url=http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com

windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第3张图片
接下去在打开Anaconda Prompt输入,命令pip install tensorflow-gpu==1.15.0即可很快的完成安装
windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第4张图片
在测试时,发生了错误如下:

windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第5张图片

Traceback (most recent call last):
  File "D:\Anaconda3\lib\site-packages\numpy\core\__init__.py", line 24, in <module>
    from . import multiarray
  File "D:\Anaconda3\lib\site-packages\numpy\core\multiarray.py", line 14, in <module>
    from . import overrides
  File "D:\Anaconda3\lib\site-packages\numpy\core\overrides.py", line 7, in <module>
    from numpy.core._multiarray_umath import (
ImportError: DLL load failed: 找不到指定的模块。

看着是numpy版本发生错误,于是我降低了版本:pip install --user numpy==1.16
windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第6张图片
由于一些不确定性我本身也不了解,一开始我安装1.14的版本出现了很多错误,比如ImportError: DLL load failed: 找不到指定的模块的问题
历程是直接pip install tensorflow-gpu,但是这个直接给你装2.2的新版本,由于想用旧版本,于是又pip uninstall tensorflow-gpu删除掉,换了博客上面说的1.15最终成功。
windows10+Nvidia GeForce GTX 1070+安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0_第7张图片

你可能感兴趣的:(配置)