Skip to content

Instantly share code, notes, and snippets.

@akhan4u
Created April 3, 2019 18:31
Show Gist options
  • Save akhan4u/bb44b2609d0c13458b94158c9c03951a to your computer and use it in GitHub Desktop.
Save akhan4u/bb44b2609d0c13458b94158c9c03951a to your computer and use it in GitHub Desktop.
Verify the setup of nvidia drivers for GPU.
root@gpu-01:~# python
Python 3.6.8 |Anaconda custom (64-bit)| (default, Dec 30 2018, 01:22:34)
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from tensorflow.python.client import device_lib
>>> print(device_lib.list_local_devices())
2019-04-03 19:09:01.537417: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
2019-04-03 19:09:02.180944: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:898] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
2019-04-03 19:09:02.181307: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1344] Found device 0 with properties:
name: GeForce RTX 2080 Ti major: 7 minor: 5 memoryClockRate(GHz): 1.545
pciBusID: 0000:01:00.0
totalMemory: 10.73GiB freeMemory: 10.57GiB
2019-04-03 19:09:02.181319: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1423] Adding visible gpu devices: 0
2019-04-03 19:09:02.376864: I tensorflow/core/common_runtime/gpu/gpu_device.cc:911] Device interconnect StreamExecutor with strength 1 edge matrix:
2019-04-03 19:09:02.376897: I tensorflow/core/common_runtime/gpu/gpu_device.cc:917] 0
2019-04-03 19:09:02.376905: I tensorflow/core/common_runtime/gpu/gpu_device.cc:930] 0: N
2019-04-03 19:09:02.377240: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1041] Created TensorFlow device (/device:GPU:0 with 10212 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:01:00.0, compute capability: 7.5)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 1452501647544097748
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 10709018215
locality {
bus_id: 1
links {
}
}
incarnation: 15364617097062970051
physical_device_desc: "device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:01:00.0, compute capability: 7.5"
]
>>> exit ()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment