前言
今年实验室配了两个2万+的设备,显卡Quardo M4000 8G 很强了。我之前还用tensorflow cpu 训练数据,慢的我都怀疑人生了。于是便安装了gpu版本。速度66的。
本文是在Win10环境下安装Anaconda3 5.0.1(64-bit),CUDA8.0,CUDNN-8.0-windows10-x64-v6.0,看了好多文章,折腾了差不多一下午和一晚上,发现这篇文章总结的比较全面,但还是有其他的问题。
问题一
CUDA的环境变量的设置:
在环境变量Path加入一下三个,第一个很重要(很多网站都忽略了!!!)
- %CUDA_PATH%\extras\CUPTI\libx64;
- %CUDA_PATH%\bin;
- %CUDA_PATH%\lib\x64;
问题二
对于CUDNN的安装,详见文章《如何安装CUDNN》,已经介绍的很清楚了。
问题三
Win10,ANACONDA3(64-bit),Python3.6.2。ANACONDA Prompt中不能用pip命令安装包,并且是在安装了TensorFlow后才发生的。
解决办法:
输入conda install -c anaconda html5lib
等待执行完后,发现pip安装指令可以成功执行。
这个问题参考文章点这里