您的位置:首页 > 其它

tensorflow及GPU加速在win10下的安装教程

2017-01-14 11:55 841 查看
本文起源于谷歌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://stackoverflow.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[/code] 
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl[/code] 但如果只输入这些,会报出
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl[/code] 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[/code] 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[/code] 
C:\> pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.1-cp35-cp35m-win_amd64.whl[/code] 
                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息