anaconda 创建pytorch环境

https://download.pytorch.org/whl/torch_stable.html

conda create -n pytorch python=3.7

pip install xxxxx.whl

安装完成后进行测试是否安装成功

python
import torch 
import torchvision
print(torch.__version__)
print(torchvision.__version__)

检测你的GPU是否可用（cuda是否可用）

import torch
print(torch.cuda.is_available())
输出为True则表示成功

是否检测到cuda版本

print(torch.version.cuda)