查看tensorflow,pytorch是否可用GPU

2022-09-28 09:09:15
  • tensorflow
import tensorflowas tfprint(tf.test.is_gpu_available())
  • pytorch
import torchprint(torch.cuda.is_available())
  • 作者：AI搬砖小能手
  • 原文链接：https://blog.csdn.net/weixin_42076509/article/details/112115472
    更新时间：2022-09-28 09:09:15