Swim-Transformer环境配置

1、下载Swim-Transformer源码到Linux服务器

git clone https://github.com/microsoft/Swin-Transformer.git
cd Swin-Transformer

2、创建并进入虚拟环境

conda create -n swin python=3.7 -y
conda activate swin

3、安装pytorch==1.7.1 cuda==10.2

pip 安装

pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2  -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

查看torch是否安装成功

import torch

print(torch.__version__)

print(torch.cuda.is_available())

显示True说明torch安装成功

4、Install timm==0.3.2

pip install timm==0.3.2 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

5、Install Apex

git clone https://github.com/NVIDIA/apex
cd apex

python setup.py install
6、Install other requirements:

pip install opencv-python==4.4.0.46 termcolor==1.1.0 yacs==0.1.8 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

7、数据集结构

imagenet
├── train
│   ├── class1
│   │   ├── img1.jpeg
│   │   ├── img2.jpeg
│   │   └── ...
│   ├── class2
│   │   ├── img3.jpeg
│   │   └── ...
│   └── ...
└── val├── class1│   ├── img4.jpeg│   ├── img5.jpeg│   └── ...├── class2│   ├── img6.jpeg│   └── ...└── ...

8、代码运行—训练模型

python -m torch.distributed.launch --nproc_per_node 4 --master_port 12345 main.py --cfg configs/swin_tiny_patch4_window7_224.yaml --data-path /home/zzk/dataset/zzk/swim_data/images --batch-size 64

其中 nproc_per_node 为GPU数量，可以设置为自己拥有的数量，不修改的话也可以，忽略警告即可；master_port 为端口号，如果此处报错 RuntimeError: Address already in use ，可以更改为其他空闲端口，如23467，88888等。

Swim-Transformer环境配置相关推荐

Swim Transformer代码环境配置纪录
Swim Transformer使用的代码:https://github.com/FZfangzheng/Swin-Transformer-Semantic-Segmentation-Without- ...
【20210122期AI简报】保姆级深度学习环境配置指南、寒武纪首颗AI训练芯片亮相...
导读:本期为 AI 简报 20210122 期,将为您带来 9 条相关新闻,新增论文推送,在文章底部,祝各位牛年大吉~ 本文一共 3700 字,通篇阅读结束需要 7~10 分钟 1. 保姆级教程:深度 ...
Astra相机的ROS开发环境配置与使用
Astra相机的ROS开发环境配置与使用一.Astra相机简介二.搭建ROS工作空间 1.下载ROS驱动功能包 2.安装依赖 3.配置工作空间三.配置设备节点 1.固定设备串口号 2.获取设备信 ...
YOLOX全文翻译及环境配置
YOLOX: Exceeding YOLO Series in 2021 声明:作者翻译论文仅为学习,如有侵权请联系作者删除博文,谢谢! 论文地址:https://arxiv.org/abs/2107 ...
Stable Diffusion - Stable Diffusion WebUI 图像生成工具的环境配置
欢迎关注我的CSDN:https://spike.blog.csdn.net/ 本文地址:https://spike.blog.csdn.net/article/details/131528224 S ...
QT5.3 + vs2012 + box2d环境配置
安装vs 和 qt 下载vs 2012,傻瓜式安装,但安装时间可能比较长.(其中一定选择"Visual C++"选项) 下载Qt 5.3,选择适合自己电脑的版本下载安装,我选的是q ...
YOLOv4 资源环境配置和测试样例效果
YOLOv4 资源环境配置和测试样例效果基本环境:cuda=10.0,cudnn>=7.0, opencv>=2.4 一.下载yolov4 git clone https://githu ...
cmd怎么实现Java你好_java环境配置以及如何在cmd窗口运行java代码
对于初学java的人来说,电脑的环境配置也许会让你头疼,但只要你认真一些学习,相信对你来说都是OK的啦~ 首先回到桌面,选择我的电脑,单击右键属性,进入高级系统设置,点击环境变量设置.用户变量选择Te ...
Springboot使用Maven Profile和Spring Profile进行多环境配置
Springboot使用Maven Profile和Spring Profile进行多环境配置目的在实际的项目上,一般会分三种环境dev.test.prod来方便我们的开发和部署,要求我们在开发的 ...
linux 编译opencl,OpenCL编译环境配置（VS+Nvidia）
英伟达的显卡首先要下载安装CUDA开发包,可以参考这里的步骤: VS2015编译环境下CUDA安装配置安装好CUDA之后,OpenCL的配置就已经完成了80%了,剩下的工作就是把OpenCL的路 ...

Swim-Transformer环境配置

Swim-Transformer环境配置相关推荐

最新文章

热门文章