最近在windows上跑深度学习模型,安装个cuda和cudnn用了好长时间,分享一下自己踩过的坑
1.去官网下载合适版本的cuda:cuda8.0,下载过9.0和9.1发现在windows7上还缺少很多组件,所有选用了cuda8.0
下载的时候最好下载local版本(完整版),这样省去在安装时候要链接外网,发生突然安装失败。
2.在到官网注册账号,并且下载cuda8.0对应的cudnn版本:cudnn6
注册账号的时候遇到很多坑,考虑的下载的方便我上传了百度云:百度云链接,密码:wogk
3.解压下载好的cudnn,把文件夹lib,bin和include里的内容,复制到对应的cuda文件夹下:
4.在anaconda里面创建虚拟环境:
$conda create --name tensorflow-gpu python=3.5 anaconda#python对应版本最好选3.5
5.启动Anaconda虚拟环境
$nvcc -V##查看cuda是否安装好$conda info —envs##查看目前多少虚拟环境$activate tensorflow-gpu
6.安装tensorflow GPU版本
$pip install tensorflow-gpu==1.4.0###该版本是对应cuda8.0
7.安装keras
$pip install keras
8.测试
import tensorflow as tfimport timesize=500W = tf.random_normal([size, size],name='W')X = tf.random_normal([size, size],name='X')mul = tf.matmul(W, X,name='mul')sum_result = tf.reduce_sum(mul,name='sum')with tf.Session() as sess:result = sess.run(sum_result)print('result=',result)
cmd下显示:
发现device有/gpu:0
注:发现在下载的时候出问题,连不上网络
解决:去用户目录下打开.condarc,删除-default一行,在增加清华源
补充虚拟环境操作:
//下面是创建python=3.6版本的环境,取名叫py36conda create -n py36 python=3.6 删除环境conda remove -n py36 --all激活环境//下面这个py36是个环境名activate py36退出环境deactivate
如果觉得《windows7下anaconda配置cuda8.0+cudnn6+tensorflow-gpu》对你有帮助,请点赞、收藏,并留下你的观点哦!