import tensorflow as tf print(tf.test.is_gpu_available()) ... PyTorch与TensorFlow测试方法类似,都有GPU测试接口。PyTorch的GPU测试代码如下:
確定! 回上一頁