软件清单:
安装环境:Windows10_64位
Python:3.6
TensorFlow:1.8.0
CUDA:CUDA9.0
cuDNN:cuDNN7.0
安装过程
1.安装python
建议安装Anaconda科学计算平台来管理你的python环境,里面自带了很多模板,下载很简单,直接下一步就好,这里就不赘述。
(若之前有其它的python版本,最好卸载了,以免不必要的麻烦)
官方下载:https://www.anaconda.com/download/
2.安装TensorFlow
安装好上述的python3.6环境后,在cmd中
输入:pip install tensorflow-gpu==1.8.0
这里默认的是下载最新版本的,本次安装是2018年7月8日下载,版本号是1.8.0。(注意,tensorflow有多个版本,可能会有多种安装方式,一定要找到对应的cuda和cuDNN版本下载,可以参照这里的安装版本)
3.安装cuda
安装cuda的时候一定要安装与当前tensorflow版本相匹配的cuda版本,这里安装的cuda版本应该是9.0的。
官方地址:https://developer.nvidia.com/cuda-toolkit-archive
(双击就可以进行安装,注意安装的时候记得关掉杀毒软件,比如360!之前一直安装失败,花了很多时间就是因为被它拦截了,一直下一步下载就好)4.安装cuDNN7
安装cuDNN库文件,这里可能需要填一个小问卷,随便填就好,下载的是一个压缩包。(下载之前需要注册账号,注册过程相当简单)
下载地址:https://developer.nvidia.com/rdp/cudnn-download
(找到对应的版本,最好使用7.0版本的,比较稳定吧)
5.关联CUDA与cuDNN
解压cuDNN压缩包后,把解压后的文件分别复制粘贴到cuda对应的文件夹里。
(到此就完成了所有的配置了)6.测试
直接用cmd进入python命令行界面
输入:import tensorflow
sess = tensorflow.Session()
就可以看到如下信息。
(可以看到显卡信息940M。。。。。。很垃圾的显卡,将就用吧)








网友评论