swin-transformer 目标检测 环境配置

我的显卡是4090 算力 8.6，pytorch版本不能太低，否则报错算力和本机不符

算力查看Compare Current and Previous GeForce Series of Graphics Cards | NVIDIA

我的版本：
python 3.8  
pytorch 1.12.1   
cuda: 11.3    
mmcv-full = 1.7.0   
mmdet = 2.25.3

1、修改源码中的版本设置

swin-transformer源码安装的mmcv和mmdet版本比较低,需要改一下，

改成你需要安装的版本

mmdet 和 apex下载源码安装 先下载好源码 然后解压后放到虚拟环境里，下面给出了下载地址

2、安装vs2019

下载vs 2019 并安装 并安装c++ 后续安装的时候要用到 c++编译库

Visual Studio 2019 版本 16.11 发行说明 | Microsoft Learn

装c++

3、然后配置vs2019的环境变量

根据你安装vs2019的路径去配，（一定要在安装了c++之后）

4、创建一个虚拟环境：

conda create -n swin_mm python=3.8 -y

5、安装pytorch:

pytorch 1.12.1 cuda: 11.3 mmcv-full = 1.7.0 mmdet = 2.25.3

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

6、安装mmcv（要和你的cuda和pytorch版本对应，不然容易出bug)

pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.12/index.html

7、安装mmdet

下载地址：mmdetection 2.25.3下载)

把下载好的mmdet源码解压到文件夹后放到conda虚拟环境swin_mm的目录下

可以先注释一个包，这个不装也行，

把上面那个改了之后再进行安装，不然容易报错安装失败

cd 命令将路径切到环境目录下mmdet源码文件的路径下

cd F:/Anaconda3/envs/swin_mm/mmdetection-2.25.3/
pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

python setup.py develop

如果上面安装pycocotools报错， 可以用下面的重新安装一波
conda install -c conda-forge pycocotools

8、安装apex

在 github 下载源码 apex下载 到本地文件夹

把该文件夹复制到虚拟环境中，

然后再切换到该路径

cd F:/Anaconda3/envs/swin_mm/apex/

输入命令： python setup.py install 等待安装完成即可

python setup.py install 

9、测试

下载好权重文件，然后在项目根目录下创建一个checkpoints文件夹，把你的权重文件放进去

然后就可以进行测试了

测试的结果：

算力7.x版本的可以参考这篇： Windows10系统下swin-transformer目标检测环境搭建_win10如何配置swin-transformer_努力毕业的W的博客-CSDN博客