一、电脑环境:
win10系统,lenovo拯救者y720,显卡gtx1060
二、安装步骤:
1.安装annacoda,我下载的是2.7版本,但是好像也可以设立python3.5的编译环境。
地址:https://www.anaconda.com/download/#windows
具体步骤不说了,很多教程都有,也很简单
2.打开anacoda prompt,创建新的环境来安装tensorflow-gpu
create tfgpu
打开新创建的环境
activate tfgpu
安装tensorflow,使用清华的镜像源,并设置搜索时显示原地址,最后搜索tensorflow-gpu
conda config --add channels 'https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/'
conda config --set show_channel_urls yes
anaconda search -t conda tensorflow-gpu
接下来选择一个源地址,我选的是aaronzs。这一步是显示安装方式
anaconda show aaronzs/tensorflow-gpu
可以跳过上一步,直接输入安装命令
conda install --channel https://conda.anaconda.org/aaronzs tensorflow-gpu
显示success以后就安装成功了,tensorflow的版本可以在anacoda navigator里面查到
3.安装cuda
要根据自己的gpu型号和tensorflow型号选择对应的cuda,对于tensorflow的型号一定要看清楚!!!不然cuda用不了,重装很麻烦,我开始下的cuda9.1结果用不了,只好卸载重装。我的tensorflow是1.9.0,因此我选的是cuda9.0,地址如下
https://developer.nvidia.com/cuda-90-download-archive
安装后进行测试,成功即可
4.安装cudnn
直接下载,然后解压,将三个文件替换到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下
下载地址如下,要下cudnn9.0版。
https://developer.nvidia.com/rdp/cudnn-download
5.下载其余python的库文件,如cv2,在anacoda prompt里面输入
conda install --channel https://conda.anaconda.org/menpo opencv3
即可,其他matplotlib更简单,直接pip install即可