Skip to content

GitLab

Commit 9fa98442 authored by dcuai's avatar dcuai
Browse files

更新dtk24.04.1镜像

parent ddf7213c
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 10 deletions
README.md
View file @ 9fa98442
...@@ -25,12 +25,11 @@ mv mmpose-rtmo_pytorch mmpose # 去框架名后缀 ...@@ -25,12 +25,11 @@ mv mmpose-rtmo_pytorch mmpose # 去框架名后缀
### Docker（方法一） ### Docker（方法一）
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk23.10-py38 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
# <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：229ce1daddf6 # <your IMAGE ID>为以上拉取的docker的镜像ID替换
docker run -it --shm-size=32G -v $PWD/mmpose:/home/mmpose -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name rtmo <your IMAGE ID> bash docker run -it --shm-size=32G -v $PWD/mmpose:/home/mmpose -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name rtmo <your IMAGE ID> bash
cd mmpose cd mmpose
pip install -r requirements.txt pip install -r requirements.txt
pip install mmcv-2.0.1+gite2f0eed.abi0.dtk2310.torch1.13-cp38-cp38-manylinux2014_x86_64.whl # dcu版mmcv，可从光合社区下载。
pip install -v -e . # 安装mmpose=1.3.0 pip install -v -e . # 安装mmpose=1.3.0
# 训练时会自动下载目标检测权重到项目默认指定位置：yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth # 训练时会自动下载目标检测权重到项目默认指定位置：yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth
``` ```
...@@ -38,10 +37,9 @@ pip install -v -e . # 安装mmpose=1.3.0 ...@@ -38,10 +37,9 @@ pip install -v -e . # 安装mmpose=1.3.0
``` ```
cd mmpose/docker cd mmpose/docker
docker build --no-cache -t rtmo:latest . docker build --no-cache -t rtmo:latest .
docker run --shm-size=32G --name rtmo -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../mmpose:/home/mmpose -it rtmo bash docker run --shm-size=32G --name rtmo -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../mmpose:/home/mmpose -it rtmo:latest bash
# 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。 # 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。
cd mmpose cd mmpose
pip install mmcv-2.0.1+gite2f0eed.abi0.dtk2310.torch1.13-cp38-cp38-manylinux2014_x86_64.whl # dcu版mmcv，可从光合社区下载。
pip install -v -e . # 安装mmpose=1.3.0 pip install -v -e . # 安装mmpose=1.3.0
# 训练时会自动下载目标检测权重到项目默认指定位置：yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth # 训练时会自动下载目标检测权重到项目默认指定位置：yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth
``` ```
...@@ -49,11 +47,11 @@ pip install -v -e . # 安装mmpose=1.3.0 ...@@ -49,11 +47,11 @@ pip install -v -e . # 安装mmpose=1.3.0
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
- https://developer.hpccube.com/tool/ - https://developer.hpccube.com/tool/
``` ```
DTK驱动:dtk23.10 DTK驱动:dtk24.04.1
python:python3.8 python:python3.10
torch:1.13.1 torch:2.1.0
torchvision:0.14.1 torchvision:0.16.0
apex:0.1 apex:1.1.0
mmcv:2.0.1 mmcv:2.0.1
# cd mmpose & pip install -v -e . # 安装mmpose=1.3.0 # cd mmpose & pip install -v -e . # 安装mmpose=1.3.0
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment