Tensorflow、Keras、Pytorch判断是否可用GPU加速
# confirm TensorFlow sees the GPUfrom tensorflow.python.client import device_libprint(device_lib.list_local_devices())# confirm Keras sees the GPUfrom keras import backendprint(backend.tensorfl...
原创
2019-12-06 22:30:28 ·
635 阅读 ·
0 评论