cuda、cudnn和torch安装的一系列

import torch # 导入PyTorch库
	
print(torch.__version__) # 打印当前PyTorch库的版本号
print(torch.cuda.is_available()) # 打印系统是否可用CUDA（即GPU是否可用）
print(torch.version.cuda) # 打印PyTorch安装时使用的CUDA版本
print(torch.cuda.device_count()) # 打印系统中可用的GPU数量
print(torch.cuda.current_device()) # 打印当前正在使用的GPU的索引
print(torch.cuda.get_device_name()) # 打印当前GPU的名称
print(torch.backends.cudnn.is_available()) # 打印系统是否可用CuDNN
print(torch.backends.cudnn.enabled) # 打印CuDNN是否启用
print(torch.backends.cudnn.version()) # 打印了当前CuDNN库的版本号

是转载的，但是搜不到原始文章了，特别提醒一下