Skip to content

GitLab

Commit 69bb1b2c authored by shantf's avatar shantf
Browse files

update README.md and dockerfile

parent 8d46e8a9
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
README.md
View file @ 69bb1b2c
...@@ -14,8 +14,8 @@ RTMDet 将 CSPDarkNet 作为基线，并使用同样构建单元组成的 CSPPAF ...@@ -14,8 +14,8 @@ RTMDet 将 CSPDarkNet 作为基线，并使用同样构建单元组成的 CSPPAF
### Docker（方法一） ### Docker（方法一）
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.8
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v /opt/hyhal:/opt/hyhal:ro --name docker_name imageID bash
cd /path/workspace/ cd /path/workspace/
pip install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/ pip install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/
``` ```
...@@ -23,17 +23,17 @@ pip install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/ ...@@ -23,17 +23,17 @@ pip install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/
``` ```
cd ./docker cd ./docker
docker build --no-cache -t mmdet:3.0 . docker build --no-cache -t mmdet:3.0 .
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v /opt/hyhal:/opt/hyhal:ro --name docker_name imageID bash
cd /path/workspace/ cd /path/workspace/
``` ```
### Anaconda（方法三） ### Anaconda（方法三）
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： https://developer.hpccube.com/tool/ 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： https://developer.hpccube.com/tool/
``` ```
DTK软件栈: dtk23.04.1 DTK软件栈: dtk24.04.1
python: python3.8 python: python3.8
torch: 1.13.1 torch: 2.1.0
torchvision: 0.14.1 torchvision: 0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1
mmcv: 2.0.0 mmcv: 2.0.1+das1.1.gite58da25.abi1.dtk2404.torch2.1.0
``` ```
......
docker/Dockerfile
View file @ 69bb1b2c
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.8
RUN source /opt/dtk/env.sh RUN source /opt/dtk/env.sh
RUN pip3 install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/ RUN pip3 install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/
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