Tensorflow-gpu不适用于pycharm,但可用于Windows 10命令提示符

时间:2019-02-22 17:57:29

标签: python tensorflow pycharm gpu

我已经通过pip安装下载了python 3.6.6,CUDA版本9,cuDNN 7.5和tensorflow-gpu 0.12。在cmd中,输入以下命令后,它会为我提供正确的输出:

import tensorflow as tf
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))

我在cmd中得到以下输出:

enter image description here

当我尝试检查tensorflow是否也在pycharm中的gpu上运行时,我在运行相同的代码后得到了这个提示:

2019-02-22 17:Device mapping: no known devices.
56:38.622466: I tensorflow/core/common_runtime/direct_session.cc:307] 
Device mapping:

我创建了一个虚拟环境,并选择python 3.6作为解释器。安装了tensorflow 0.11和tensorflow-gpu 0.11(以确保这些版本与CUDA 9和cuDNN 7.5兼容)。 我尝试卸载和安装这些软件包,但没有帮助。 路径已添加如下: enter image description here

以及pycharm中的项目结构:

enter image description here

我对在此还要检查的内容感到困惑。感谢您的帮助。

0 个答案:

没有答案