PyTorch安装

PyTorch快速上手

PyTorch安装

这里选择配置的环境版本如下

环境	版本
PyTorch	1.10.1
CUDA	11.1
Python	3.9.7

其他的历史版本可以去官网搜索相应的安装命令，链接如下

Previous PyTorch Versions | PyTorch

首先创建虚拟环境

conda create -n pytorch-1.10.1 python=3.9.7

检查当前激活的环境列表，看到有pytorch-1.10.1即可（可选）

conda env list

激活当前环境

conda activate pytorch-1.10.1

执行安装命令

pip install torch==1.10.1+cu111 torchvision