1、CUDA是否可用
print(tf.test.is_built_with_cuda())
2、查看GPU是否可用
print(tf.test.is_gpu_available())
3、查看所有GPU列表
print(tf.config.list_physical_devices('GPU'))
1、CUDA是否可用
print(tf.test.is_built_with_cuda())
2、查看GPU是否可用
print(tf.test.is_gpu_available())
3、查看所有GPU列表
print(tf.config.list_physical_devices('GPU'))