计算机系统和显卡配置
本机系统windows10 64位、显卡英伟达 GTX1060
安装tensorflow的版本选择
注意:这里需要特备留意cudnn的版本必须与tensorflow-gpu的版本匹配,tensorflow的版本为1.3.0。Anaconda安装后可使用命令conda list查看。
开始安装
(1)安装Anaconda
使用pip -V查看版本,低于9.0则需要输入python -m pip install -U pip更新。
推荐大家使用清华源镜像,这样以后使用起来比较方便,也避免了一些下载速度安装出错的各种问题。在命令提示符里输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes (当前cmd配置conda)
(2)安装cuda
点击exe文件,使用默认选项安装即可。
(3)解压cudnn到当前目录,得到cuda文件夹
把这三个文件夹里的文件分别放入安装好的cuda的目录,比如:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
路径下的对应文件夹里面,注意把lib里面x64里的文件复制过去,不要把文件夹复制过去,一个一个文件打开对应复制过去。
(4)安装tensorflow-gpu
pip install tensorflow-gpu==1.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
注意:Tensorflow的版本一定与安装的cuda和cudann匹配才行。
(5)测试是否安装成功
使用import tensorflow as tf,出现图中的结果即是安装成功!
领取专属 10元无门槛券
私享最新 技术干货