tensorflow-gpu-pycharm无法识别GPU

时间:2018-12-11 18:37:59

标签: tensorflow

我已经导入了tensorflow gpu,但是pycharm无法识别

详细信息:

IDE -pycharm
GPU- grid p40-1Q
cuda - 8
cudnn - 7.1
python - 3.5

我收到此消息:

enter image description here

1 个答案:

答案 0 :(得分:0)

您检查过

  

nvidia-smi

,然后查看您的图形驱动程序是否正常工作?另外,您是否检查过tensorflow-gpu的版本?在https://www.tensorflow.org/install/source#tested_source_configurations

中检查兼容的配置

我在Windows 10的pycharm上尝试了以下配置,并且有效!!

  1. 带有python 3.6.8的Anaconda导航器
  2. CUDA 9.0
  3. CUDNN 7.1
  4. Nvidia GeForce 1050 Ti GPU

按照https://stackoverflow.com/a/51307381/2562870

中的说明进行操作