tensorflow-gpu 版本比较难安装,几次安装成功后发现其实用的还是cpu,不建议用conda安装,如果你不是非要使用gpu的话,conda安装确实最快,使用以下命令即可
#注意conda版本中python要为3.6版本,建议安装conda4.6.12
conda install tensorflow-gpu==1.2.1 keras
但是发现这样安装之后还是使用的cpu,可以使用以下代码测试
import numpy
import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))
```python
#gpu版本输出应为
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce GTX 960M, pci bus id: 0000:01:00.0, compute capability: <

最低0.47元/天 解锁文章

被折叠的 条评论
为什么被折叠?



