pip 安装pyTorch
pip install torch1.3.0+cu92 torchvision0.4.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html --no-cache-dir
安装的torch版本, cuda版本, torchvision版本 需要对应
官网地址
pip库安装临时源: -i https://pypi.tuna.tsinghua.edu.cn/simple
测试gpu
import torch
torch.cuda.is_available()
# 输出True则安装成功