测试cuda是否安装成功命令

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是否生效