Check GPU
Wed Jul 27 2022 11:47:31 GMT+0000 (Coordinated Universal Time)
Saved by
@mnis00014
import tensorflow as tf
print(tf.__version__)
gpu_info = !nvidia-smi
gpu_info = '\n'.join(gpu_info)
if gpu_info.find('failed') >= 0:
print('Not connected to a GPU')
else:
print(gpu_info)
content_copyCOPY
Comments