首先打开anaconda prompt，激活对应虚拟环境。

导入torch并获取对应版本

import torch
torch.__version__

 

导入torchvision并获取对应版本

import torchvision
torchvision.__version__

检查cuda是否可用 

torch.cuda.is_available()

获取CUDA设备数

torch.cuda.device_count()

获取当前cuda设备id

torch.cuda.current_device()

获取CUDA设备名称

torch.cuda.get_device_name(0)