安装tensorflow-gpu需要另外安装cuda,cudnn,众所周知
我用的MX150这款显卡
可以右键查看cuda核心:
我这里是384
参考:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
用不了cuda9.0,只能用cuda8.0
在anaconda navigator里面下载的tensorflow-gpu会自动下载最新版本cuda,目前都到10.几了,所以这里要降低版本
测试一下:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
能用!嘻嘻