Win10_64位下搭建TensorFlow-GPU版本的环境
标签 : 深度学习
软件清单:
- Python : 3.6.2 版本
- TensorFlow : 1.4 版本
- CUDA : CUDA8_win10 版本
- cuDNN : cuDNN6.0_win10 版本
本文末尾会提供本次博文使用到的所有安装文件的百度云盘分享链接。
安装过程:
1. 安装Python
两种方式 :
到官网上下载安装包进行安装
下载地址 : https://www.python.org/downloads/windows/使用Anaconda科学计算平台来管理你的Python环境
下载地址 : https://www.anaconda.com/download/
由于Python和Anaconda的安装不是本文重点,故此处略去安装过程。如有需要,请自行查阅相关博文。
2. 安装TensorFlow
#GPU版本
pip3 install –upgrade tensorflow-gpu#CPU版本
pip3 install –upgrade tensorflow
请注意上面命令中upgrade前面是两个横杠。
这边本人使用pip3安装TensorFlow的时候过程非常顺利,读者如果出现问题的话可能是网络或者python环境的问题吧,毕竟本人的渣网速也可以顺利安装成功。
安装好tensorflow后,我们可以先尝试在python命令行中使用tensorflow模块,看看会出什么问题。
可以看到提示我们找不到’cudart64_80.dll’,需要下载并安装CUDA8.0。
接下来我们进行CUDA8的下载和安装。
3. 安装CUDA8
下载地址 : https://developer.nvidia.com/cuda-downloads
访问NVIDIA官网找到CUDA下载页面 , 点击L