2.安装conda python库

centos 显卡驱动、cuda、cudnn安装参考地址：

1.centos7 安装显卡驱动、cuda、cudnn-CSDN博客

1.安装anaconda

步骤1：下载

下载地址：

Index of /

下载版本

步骤2：运行

sh Anaconda3-2021.05-Linux-x86_64.sh

一路执行回车，然后选择yes，最后安装完成

步骤3：测试

如果没有出现anacoda，则添加环境变量

vim ~/.bashrc

添加anaconda的地址

使环境变量生效，source ~/.bashrc

再执行python命令测试

新建空间：

conda create -n swin python=3.8

conda activate swin

步骤4：安装包

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

由于在线安装总出错，建议先将包下载下来之后，再安装。由于torch，依赖numpy，所以要先安装numpy。

pip install numpy -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

pip install torch-1.8.0+cu111-cp38-cp38-linux_x86_64.whl --trusted-host pypi.douban.com

安装torchvision依赖于pillow

pip install pillow -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

pip install torchvision-0.9.0+cu111-cp38-cp38-linux_x86_64.whl --trusted-host pypi.douban.com

pip install torchaudio-0.8.0-cp38-cp38-linux_x86_64.whl  --trusted-host pypi.douban.com

步骤5：测试

返回True表示成功。

步骤6：安装其他的包

pip install  cython matplotlib opencv-python