前言
如果已经安装了pytorch版本，对应跳过步骤一，直接安装步骤二即可
注意：依赖包的配置需要根据pytorch版本号以及cuda的版本号，版本号要适配，否则部分函数无法使用
1. 安装pytorch
基于代码安装适配的pytorch版本
通过官网进行安装：pytorch下载官网，此处版本根据代码需要，具体格式如下
- 命令如下（linux）：
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch