记录下按照cuda的过程(win10+python3.7+cuda10.0+cudnn7.6.5+TensorFlow1.13.2)

今天按照一下午,终于算是搞好了,记录一下安装过程。

1.更新显卡驱动(驱动下载)。

在设备管理器中找到显卡对于的型号,然后去上面地址里面选择相应的驱动下载

2.下载cuda(点这下载),cuda的版本和Python,tf,cudnn都要对应。对应表网上能搜到。

这里注意每个小版本是不一样的,需要安装10.0就不能选择10.1!
记录下按照cuda的过程(win10+python3.7+cuda10.0+cudnn7.6.5+TensorFlow1.13.2)_第1张图片
这个界面是选择操作系统的型号的!!!我好几次当成了选择cuda的型号,明明是win10选了个8.1.。。。,注意注意!!!

这里已经自动给cuda的bin目录加入环境变量了,可以用nvcc -V测试一下(这是查看版本号的)

3.下载cudnn(点这下载),找到对应cuda的最新的一个版本(对应版本应该都行)。

这个一个zip,下载后解压,将对于的文件移动到cuda的对于目录。
记录下按照cuda的过程(win10+python3.7+cuda10.0+cudnn7.6.5+TensorFlow1.13.2)_第2张图片

4.安装tf。tf1.5之前cpu和gpu是2个版本,pip install tensorflow安装的是cpu版,pip install tensorflow-gpu 安装的是gpu版本,只需要安装一个就好,都装了默认启动gpu版。

有时候会出现tf的部分模块无法导入,这可能是tf包损坏了,卸载重装,还有时候,提示说本地dll无法加载,这种情况一般是cuda+tf+cudnn那一个没有对应。最后重启(我的重启才好,有的机子貌似不用)。

import tensorflow as tf

a=tf.constant(3)
b=tf.constant(6)
with tf.Session() as sess:
    print(sess.run(a+b))

用这个代码测试是否安装成功。
记录下按照cuda的过程(win10+python3.7+cuda10.0+cudnn7.6.5+TensorFlow1.13.2)_第3张图片
sess.run的结果必须要print出来,cpu版本不print也能输出,但gpu版必须要print,第一次测试时候没写print,最后结果没输出,懵逼好久。

你可能感兴趣的:(机器学习)