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