下关网站建设,广告推广营销网站,莱芜东风街,英文自助建站首先打开anaconda prompt#xff0c;激活对应虚拟环境。
导入torch并获取对应版本
import torch
torch.__version__导入torchvision并获取对应版本
import torchvision
torchvision.__version__ 检查cuda是否可用
torch.cuda.is_available() 获取CUDA设备数
torch.cuda.…首先打开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)
