import torch
torch.__version__ # 查看pytorch版本
torch.cuda.is_available() # 判断pytorch是否支持GPU加速
torch.version.cuda # 查看CUDA版本
torch.backends.cudnn.version() # 查看cuDNN版本
torch.cuda.get_device_name(0) # 查看显卡类型
cuda常用指令
于 2023-04-28 23:49:53 首次发布