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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值