欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

tensorflow-gpu1.3.0安装(win10+cuda8.0.61+cuDNN6.0)

程序员文章站 2022-07-04 13:40:26
...

其实网上的方案有很多

http://m.blog.csdn.net/JohinieLi/article/details/76960112

http://blog.csdn.net/weixin_36368407/article/details/54177380

http://blog.csdn.net/u010099080/article/details/53418159

这都是我当初安装时所参考的

我就做个大概总结

我所用到的软件(基础):

Anaconda3(python3.5) +vs2015 +cuda8.0.61+cuDNN6.0+tensorflow-gpu1.3.0

安装过程以上链接都有介绍

下面是几点心得体会:

1.下载cuda和cudnn时建议用win10自带的浏览器,我用的360不是下不了 就是奇慢。

2.一定要看好对应的版本!

之前网上的多半是 cuda8.0.44+cudnn5.0

然而当我安装的tensorflow的版本是1.3.0时就需要用cudnn6.0了

否则像我第一次就出现了一大长串错误,导致我怀疑是不是应该把cuda8.0.61换成 cuda8.0.44

最后用矩阵乘法测试下是否通车

import tensorflow as tf

a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)


tensorflow-gpu1.3.0安装(win10+cuda8.0.61+cuDNN6.0)