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)