TensorFlow_gpu是否配置ok测试

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'))

你可能感兴趣的:(tensorflow,深度学习,python)