def print_torch_test():
print("pyTorch版本:",torch.__version__) # 输出pyTorch版本
print("CUDA版本:",torch.version.cuda) # 输出CUDA版本
print("cuDNN的版本:",torch.backends.cudnn.version) # 输出cuDNN的版本
print("显卡:",torch.cuda.get_device_name(0)) # 输出显卡版本
print(torch.cuda.is_available()) # 测试GPU是否生效
测试cuda是否安装成功命令
于 2022-07-07 14:39:32 首次发布