我刚刚在机器上安装了Tensorflow的新版本,并使用tensorflow-gpu创建了一个conda环境。
在激活环境的情况下,我在命令行中运行了典型的GPU利用率代码,以检查TF是否正在使用我的GPU,这很好。
dif = np.abs(np.subtract.outer(A,B))
np.ma.array(dif, mask = A[:,None] < B).min(1).data
# array([1, 2, 3, 2, 0, 2, 3, 4])
但是,当我尝试设置PyCharm在这种环境下使用解释器时,它只是失败了。运行配置和项目配置是相同的,它们应该使用import tensorflow as tf
with tf.device('/gpu:0'):
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)
with tf.Session() as sess:
print (sess.run(c))
环境,但是在PyCharm中运行相同的检查代码时,它将失败并显示以下信息:
gpu_env
项目设置为:
运行/调试配置设置为:
这是我几个月前最后一次尝试使用的方法,但我不确定现在什么不起作用。
任何提示都会非常有帮助!谢谢!
答案 0 :(得分:0)
问题似乎是特定于版本的,在这种情况下为: 2019.1
。在解决此PyCharm错误之前,该解决方案更多是一种变通方法。
或者,您可以执行以下操作: