200字范文,内容丰富有趣,生活中的好帮手!
200字范文 > windows7下anaconda配置cuda8.0+cudnn6+tensorflow-gpu

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

时间:2019-01-24 12:46:43

相关推荐

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 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

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。