import tensorflow as tf
import torch
print(tf.test.is_gpu_available())
print(torch.cuda.is_available())
检查Tensorflow-gpu和pytorch-gpu是否可用
最新推荐文章于 2022-04-08 22:42:14 发布
import tensorflow as tf
import torch
print(tf.test.is_gpu_available())
print(torch.cuda.is_available())