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)
上一篇: 为啥iPad Pro还不支持3D Touch?有原因
下一篇: 平胸妹纸的烦恼