博客

检查Tensorflow-gpu和pytorch-gpu是否可用

  • Post author:
  • Post category:其他


import tensorflow as tf
import torch


print(tf.test.is_gpu_available())
print(torch.cuda.is_available())


版权声明：本文为weixin_41529389原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。