Skip to content

GitLab

Commit 59e7a34f authored by zzg_666's avatar zzg_666
Browse files

update

parent 5e0994da
Hide whitespace changes
Inline Side-by-side
Showing with 23 additions and 12 deletions
README.md
View file @ 59e7a34f
...@@ -22,10 +22,14 @@ mv yolo11_pytorch YOLO11 # 去框架名后缀 ...@@ -22,10 +22,14 @@ mv yolo11_pytorch YOLO11 # 去框架名后缀
### Docker（方法一） ### Docker（方法一）
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10
# <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：83714c19d308 # <your IMAGE ID>为以上拉取的docker的镜像ID替换
docker run -it --shm-size=64G -v $PWD/YOLO11:/home/YOLO11 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name yolo11 <your IMAGE ID> bash docker run -it --shm-size=64G -v $PWD/YOLO11:/home/YOLO11 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name yolo11 <your IMAGE ID> bash
cd /home/YOLO11 cd /home/YOLO11
pip uninstall numpy
pip uninstall ray
pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install ray==2.37.0
pip install -r requirements.txt # requirements.txt pip install -r requirements.txt # requirements.txt
``` ```
### Dockerfile（方法二） ### Dockerfile（方法二）
...@@ -33,22 +37,21 @@ pip install -r requirements.txt # requirements.txt ...@@ -33,22 +37,21 @@ pip install -r requirements.txt # requirements.txt
cd cd /home/YOLO11/docker cd cd /home/YOLO11/docker
docker build --no-cache -t yolo11:latest . docker build --no-cache -t yolo11:latest .
docker run --shm-size=64G --name yolo11 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../YOLO11:/home/YOLO11 -it yolo11 bash docker run --shm-size=64G --name yolo11 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../YOLO11:/home/YOLO11 -it yolo11 bash
pip uninstall numpy
pip uninstall ray
pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install ray==2.37.0
# 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。 # 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。
``` ```
### Anaconda（方法三） ### Anaconda（方法三）
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
- https://developer.sourcefind.cn/tool/ - https://developer.sourcefind.cn/tool/
``` ```
DTK驱动:dtk24.04.2 DTK驱动:dtk25.04.2
python:python3.10 python:python3.10
torch:2.3.0 torch:2.5.1
torchvision:0.18.1 torchvision:0.20.1
torchaudio:2.1.2 torchaudio:2.5.1
triton:2.1.0
flash-attn:2.0.4
deepspeed:0.14.2
apex:1.3.0
xformers:0.0.25
``` ```
`Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应。` `Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应。`
...@@ -56,6 +59,10 @@ xformers:0.0.25 ...@@ -56,6 +59,10 @@ xformers:0.0.25
2、其它非特殊库参照requirements.txt安装 2、其它非特殊库参照requirements.txt安装
``` ```
cd /home/YOLO11 cd /home/YOLO11
pip uninstall numpy
pip uninstall ray
pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install ray==2.37.0
pip install -r requirements.txt # requirements.txt pip install -r requirements.txt # requirements.txt
``` ```
......
docker/Dockerfile
View file @ 59e7a34f
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10 FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
# RUN yum update && yum install -y git cmake wget build-essential # RUN yum update && yum install -y git cmake wget build-essential
# RUN source /opt/dtk-24.04.2/env.sh # RUN source /opt/dtk-24.04.2/env.sh
# # 安装pip相关依赖 # # 安装pip相关依赖
COPY requirements.txt requirements.txt COPY requirements.txt requirements.txt
RUN pip uninstall numpy
RUN pip uninstall ray
RUN pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip install ray==2.37.0
RUN pip3 install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com RUN pip3 install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
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