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

import tensorflow as tf
import torch


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

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页