windows10+tensorflow(GPU)+python+cuda+cudnn+keras安装教程

之前安装过CPU版本,现在安装GPU版本,网上很多教程有的不完整,趁着刚刚安装完赶快记下来,供日后参考。

1.Anaconda安装
①网址:https://www.anaconda.com/download/

②选择windows

③下载
下载Anaconda

④按照步骤安装
注意:要勾选添加路径

2.CUDA和CuDNN安装
①(可有可无)查看GPU可用的型号支持
网址:https://developer.nvidia.com/cuda-gpus

②下载CUDA
一般情况下,tensorflow(GPU)1.3对应的是CUDA8.0,所以选择8.0,而不是选择安装最新版本。
网址:https://developer.nvidia.com/cuda-toolkit-archive
下载CUDA

③安装CUDA
按步骤安装,但是记住下列勾选
该图片转载于【1】

该图片转载于【1】

④下载CuDNN
网址:https://developer.nvidia.com/rdp/cudnn-download
注意:需要注册,很快
注册后,进入页面
下载CuDNN
勾选后,选择版本如图
CuDNN版本

⑤解压CuDNN,将三个文件夹替换CUDA的相应文件夹
该图片转载于【1】

该图片转载于【1】

⑥检查安装
在cmd中输入nvcc -V
该图片转载于【2】

3.安装tensorflow的环境
Anaconda3默认安装的是Python3.6,但是目前的tensorflow只支持python3.5,这个时候要重新创建一个python3.5的环境(注意:该环境命名为python35),在该环境下安装tensorflow(备注:网上还有很多教程是直接安装一个命名为tensorflow-gpu的环境,其实是一样的)。
①安装python3.5环境
运行-cmd(或直接打开anaconda prompt)
配置清华仓库镜,输入指令:
conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config –set show_channel_urls yes
输入conda create –name python35 python=3.5

②激活python35环境
激活:activate python35
取消激活:deactivate

③在该环境下安装tensorflow
先激活python35环境,再输入指令:
pip install tensorflow-gpu

④测试
打开anaconda navigator,在home标签下,applications on切换环境:python35,找到spyder,选择install,等待安装后,选择launch,进入后输入指令:
import tensorflow as tf
如果不报错,就是成功。如果报错,很可能是tensorflow-gpu的版本错误,可能装成了最新版的,如果是这样,需要进行以下步骤:
同上一样在anaconda prompt中激活环境python35,输入
pip install –upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl
安装后,就是tensorflow-gpu的1.3版本,之后再重新测试。

4.安装keras
先激活python35环境,再输入指令:
pip install keras
同样测试,打开python35环境下的spyder,输入指令:
import keras
不报错即为成功。

5.以后每次运行代码中若提示“no module named XXX”,都要去anaconda prompt中activate python35,然后pip install XXX安装即可。

6.参考:
【1】https://blog.csdn.net/weixin_39290638/article/details/80045236
【2】http://www.cnblogs.com/elroye/p/7864988.html
【3】https://blog.csdn.net/qq_34654240/article/details/80301147

猜你喜欢

转载自blog.csdn.net/u011591202/article/details/82453553
今日推荐