记录在linux中安装pytorch

1. 搭建Anaconda环境

2. 标pytorch的安装（由GPU）

2.1.首先创建自己的虚拟环境

conda create -n royolove

2.2查看自己创建的环境

conda env  list 

2.2激活royolov5房间

source activate royolov5

2.3进入pytorch官网

https://pytorch.org/

找到自己对应的pytorch,如果在这里没有找到，进入红色的位置,以我为例我下载的CUDA 10.1的环境，复制所对应的语句`。

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

输入后就安装完成了

2.4检测是否安装成功

python
import torch
print (torch.cuda.is_available())

显示True表示安装成功了

2.5如果要离线安装可以先下载自己对应的torch和torchvision的包

地址为https://download.pytorch.org/whl/torch_stable.html

要找到与自己python对应的包。