檢視tensorflow,pytorch是否可用GPU

AI搬磚小能手發表於2021-01-02
PyTorchGPU
  • tensorflow
import tensorflow as tf
print(tf.test.is_gpu_available())
  • pytorch
import torch
 print(torch.cuda.is_available())

相關文章