安装 Visual Studio 下载地址
下载完成后点击exe,勾选通用Windows平台开发和使用c++的桌面开发,直到安装完成。
右键点击NVIDIA控制面板,进入帮助->系统信息,选择组件,查看NVCUDA.dll上写的CUDA版本号
我的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=/pypi/simple/[install]trusted-host=
接下去在打开Anaconda Prompt输入,命令pip install tensorflow-gpu==1.15.0
即可很快的完成安装
在测试时,发生了错误如下:
Traceback (most recent call last):File "D:\Anaconda3\lib\site-packages\numpy\core\__init__.py", line 24, in <module>from . import multiarrayFile "D:\Anaconda3\lib\site-packages\numpy\core\multiarray.py", line 14, in <module>from . import overridesFile "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
由于一些不确定性我本身也不了解,一开始我安装1.14的版本出现了很多错误,比如ImportError: DLL load failed: 找不到指定的模块的问题
历程是直接pip install tensorflow-gpu,但是这个直接给你装2.2的新版本,由于想用旧版本,于是又pip uninstall tensorflow-gpu删除掉,换了博客上面说的1.15最终成功。