本文起源于谷歌2016年11月底发的一篇blog:
https://developers.googleblog.com/2016/11/tensorflow-0-12-adds-support-for-windows.html
宣布tensorflow正式支持windows且在其上运行NVIDIA的GPU加速。今早成功在自己的win10上安装tensorflow,下面讲一下安装的整个流程:
BTW,我的电脑显卡配置:
NVIDIA GeForce 940MX
1.安装Anaconda
由于xx(编译器什么的,没看清楚)原因,完整的tensorflow只支持python3.5,(使用Docker的情况除外)。所以Anaconda一般下载python3.5版本。
2.安装CUDA Drivers
资源在这里,选对自己电脑对应的系统即可。下载完以后点开安装,跟着安装指导一步步走下去,最好选择默认路径C盘,可以省去不必要的麻烦。(我第一次放在E盘,出了点问题)。
3.下载cuDNN - CUDA for Deep Neural Networks
这其实就是CUDA的深度神经网络支持,下载下来以后解压出来一个cuda文件夹,放到你想放的位置,然后把这个cuda文件夹下的bin目录地址放到PATH里。
如果不在这里配置环境变量的话,结束安装流程以后,import tensorflow时会出一点问题
附上当时参考的网站:
http://*.com/questions/41007279/tensorflow-on-windows-couldnt-open-cuda-library-cudnn64-5-dll
4.通过Pip安装tensorflow
谷歌官网上说的比较清楚,如何通过pip安装tensorflow的CPU版本和GPU版本。但是按照官网上的操作会出现一些问题。
举例:假如你打算安装GPU版本,则只需要在命令行输入如下命令:
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
但如果只输入这些,会报出
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
cannot remove nonexistent setuptools的错误。
正确的做法:
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
pip install --upgrade --ignore-installed https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl
一定要加上‘--ignore-installed’。
所以第四步总的来说就是在你的命令行里输入上面的代码。
5.在命令行里试运行:
可以看到所有库都加载成功,没有第三步里出现的问题。
END
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl