简体   繁体   中英

Tensorflow not running on GPU in PyCharm

I installed Tensorflow GPU and I can see it in Terminal on Ubuntu 18.04.

>>> from tensorflow.python.client import device_lib
>>> print(device_lib.list_local_devices())

And result is:

2018-08-21 16:34:49.867482: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1484] Adding visible gpu devices: 0
2018-08-21 16:34:49.867602: I tensorflow/core/common_runtime/gpu/gpu_device.cc:965] Device interconnect StreamExecutor with strength 1 edge matrix:
2018-08-21 16:34:49.867627: I tensorflow/core/common_runtime/gpu/gpu_device.cc:971]      0 
2018-08-21 16:34:49.867650: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] 0:   N 
2018-08-21 16:34:49.867979: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1097] Created TensorFlow device (/device:GPU:0 with 6863 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1070 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 10929943929948782169
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 7196478669
locality {
  bus_id: 1
  links {
  }
}
incarnation: 1784950594023049308
physical_device_desc: "device: 0, name: GeForce GTX 1070 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1"
]

But when I run it in PyCharm I get

/usr/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: numpy.dtype size changed, may indicate binary incompatibility. Expected 96, got 88
  return f(*args, **kwds)
/usr/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: numpy.dtype size changed, may indicate binary incompatibility. Expected 96, got 88
  return f(*args, **kwds)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 12856835725849494278
]
2018-08-21 16:38:54.238307: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

Process finished with exit code 0

I am new to this, so any help is greatly appreciated! Thank you.

I have fixed it. I changed Project interpreter in PyCharm and now it's work for me.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM