You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 

1.5 KiB

In [4]:
import os
os.environ['LD_LIBRARY_PATH'] = '/usr/local/cuda-9.1/targets/x86_64-linux/lib/'
from tensorflow.python.client import device_lib
In [7]:
device_lib.list_local_devices()
Out[7]:
[name: "/device:CPU:0"
 device_type: "CPU"
 memory_limit: 268435456
 locality {
 }
 incarnation: 8024912505704686755, name: "/device:GPU:0"
 device_type: "GPU"
 memory_limit: 227540992
 locality {
   bus_id: 1
   links {
   }
 }
 incarnation: 17421789173799713590
 physical_device_desc: "device: 0, name: GeForce GTX 960M, pci bus id: 0000:01:00.0, compute capability: 5.0"]