Windows 2012服务器安装GPU版TensorFlow完全攻略
一、首先,推荐用Anaconda安装
因为Anaconda本身就已经默认安装了很多常用的Python库,可以省去大量的库安装过程,并且解决兼容性问题。
Anaconda本身的安装也非常简单,搜索Anaconda官网,下载最新的Anaconda对应的exe文件,约几百兆,双击安装即可。
启动Anaconda navigator,切换到Environments选项卡,选择Open Terminal启动DOS命令窗口。如下图:
二、接着,安装GPU版Tensorflow
DOS命令行输入此命令开始安装:
pip install --upgrade tensorflow-gpu
如安装指定版本则是:
pip install --upgrade tensorflow-gpu==1.3.0
最新版是1.8版Tensorflow。
三、继续安装CUDA和cudnn依赖环境
1、安装CUDA需要先安装VISUAL STUDIO 2013以上(最好安装VS2013);
2、安装CUDA8.0和cudnn v6.0;
所有CUDA版本地址:https://developer.nvidia.com/cuda-toolkit-archive
CUDA8地址:http://developer2.download.nvidia.com/compute/cuda/8.0/secure/prod/local_installers/cuda_8.0.44_windows.exe
cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download,选择windows 7版本的cudnn v6.0 for CUDA 8.0 即可在windows 2012下正常运行。
cudnn直接解压,如解压到C:\CUDNN后,把子文件夹bin目录加入到系统变量path中即可。
至此,GPU版的TensorFlow安装及配置完毕。
注意事项:
1、安装CUDA之前必须关闭杀毒软件和360等,否则可能安装失败。
2、如果之前已安装高版本显卡驱动,安装CUDA时需要自定义并取消最下面的驱动安装,否则也会安装失败。
问题及解决办法:
ReadTimeoutError: HTTPSConnectionPool(host='pypi.python.org', port=443): Read timed out.
遇到此下载库超时错误,可命令后加参数设置超时时间
pip --default-timeout=100 install