win10,gtx1060,安装tensorflow-gpu1.9.0,cuda9.0,cudnn9.0

一、电脑环境:
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即可

你可能感兴趣的:(win10,gtx1060,安装tensorflow-gpu1.9.0,cuda9.0,cudnn9.0)