【发布时间】:2019-05-08 20:22:03
【问题描述】:
我刚刚在我的机器上安装了新版本的 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 设置为在此环境中使用解释器时,它就失败了。 Run 配置和 Project 配置是相同的,它们应该使用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
项目设置为:
并且运行/调试配置设置为:
几个月前我上次尝试过这曾经可以工作,但我不确定现在什么不工作。
任何提示都会非常有帮助!谢谢!
【问题讨论】:
-
如果是pycharm的问题,可以尝试重新缓存pycharm(文件-无效缓存/重新启动)
-
我相信你正面临着已知的 PyCharm 问题youtrack.jetbrains.com/issue/PY-35141
-
@PavelKarateev 是的,就是这样。一旦你在正确的 conda 环境中,有一个答案建议从终端启动 pycharm。
标签: python tensorflow pycharm