windows7下anaconda配置cuda8.0+cudnn6+tensorflow-gpu

最近在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 tf
import time
size=500
W = 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版本的环境,取名叫py36
conda create -n py36 python=3.6 
删除环境
conda remove -n py36 --all
激活环境
//下面这个py36是个环境名
activate py36
退出环境
deactivate

猜你喜欢

转载自blog.csdn.net/orange_littlegirl/article/details/80336863