pytorch安装教程

安装pytorch

创建conda虚拟环境，选择安装电脑CUDA版本的安装即可

• 最新版：Start Locally | PyTorch

• 历史版本：Previous PyTorch Versions | PyTorch

# CUDA 10.1
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

?验证测试：

# Pytorch测试
import torch

print(torch.__version__) # 查看Pytorch版本 
print('gpu:', torch.cuda.is_available()) # 是否可以使用GPU