diatoms_detector/List tensorflow devices.ipynb

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"]