一、在conda中新建虚拟环境
conda create -n pytorch python=3.7
二、安装
conda install pytorch torchvision cudatoolkit=10.1
pip install torch
三、测试
import torch
x = torch.rand(5, 3)
print(x)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name(0))
print(torch.cuda.current_device())