我刚刚在我的机器上安装了一个新版本的Tensorflow,我用tensorflow-gpu创建了一个conda环境。
在我的环境激活的情况下,我在命令行中运行了典型的GPU利用率代码,以检查TF是否在使用我的GPU,这很好。
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
环境,但是当在PyCharm中运行相同的检查代码时,它只会失败:
Traceback (most recent call last):
File "C:/AccesaProjects/TFProj/tf_code.py", line 1, in <module>
import tensorflow as tf
File "C:\Users\radu.gheorghiu\AppData\Local\Continuum\anaconda3\envs\gpu_env\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
File "C:\Users\radu.gheorghiu\AppData\Local\Continuum\anaconda3\envs\gpu_env\lib\site-packages\tensorflow\python\__init__.py", line 47, in <module>
import numpy as np
File "C:\Users\radu.gheorghiu\AppData\Local\Continuum\anaconda3\envs\gpu_env\lib\site-packages\numpy\__init__.py", line 140, in <module>
from . import _distributor_init
File "C:\Users\radu.gheorghiu\AppData\Local\Continuum\anaconda3\envs\gpu_env\lib\site-packages\numpy\_distributor_init.py", line 34, in <module>
from . import _mklinit
ImportError: DLL load failed: The specified module could not be found.
Process finished with exit code 1
项目设置如下:
运行/调试配置设置为:
上一次我几个月前试过这种方法时,我不知道现在有什么不管用。
任何提示都会很有帮助!谢谢!
发布于 2019-05-08 04:22:03
在本例中,问题似乎是特定于版本的:2019.1
.在这个PyCharm错误被解决之前,这个解决方案更多的是一个解决方案。
作为一种选择,您可以这样做:
https://stackoverflow.com/questions/56025758
复制相似问题