pytorch-gpu安装-win10-python3.7-anaconda3

1.安装

创建虚拟环境pytorch,pip 安装gpu版本，没有cuda就选安装cpu

conda create -n pytorh python=3.7

pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

https://pytorch.org/

2.测试

import torch    # 如正常则静默
a = torch.Tensor([1.])    # 如正常则静默
a.cuda()    # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"