Skip to content

GitLab

Commit 21defecb authored by yongshk's avatar yongshk
Browse files

Initial commit

parent dc8094a6
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
Dockerfile
View file @ 21defecb
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
COPY requirements.txt requirements.txt COPY requirements.txt requirements.txt
RUN source /opt/dtk-22.10/env.sh RUN source /opt/dtk-24.04.1/env.sh
RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' >/etc/timezone RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' >/etc/timezone
ENV LANG C.UTF-8 ENV LANG C.UTF-8
RUN pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com RUN pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
README.md
View file @ 21defecb
...@@ -21,9 +21,9 @@ transformer-XL ...@@ -21,9 +21,9 @@ transformer-XL
### Docker（方法一） ### Docker（方法一）
此处提供[光源](https://sourcefind.cn/#/main-page)拉取docker镜像的地址与使用步骤 此处提供[光源](https://sourcefind.cn/#/main-page)拉取docker镜像的地址与使用步骤
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it --network=host --name=transformer-XL --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=32G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest docker run -it --network=host --name=transformer-XL --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 -v /opt/hyhal:/opt/hyhal:ro -v /usr/local/hyhal:/usr/local/hyhal:ro image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 /bin/bash
pip install -r requirements.txt pip install -r requirements.txt
``` ```
### Dockerfile（方法二） ### Dockerfile（方法二）
...@@ -43,10 +43,10 @@ pip install -r requirements.txt ...@@ -43,10 +43,10 @@ pip install -r requirements.txt
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动：dtk22.10 DTK驱动：dtk24.04.1
python：python3.7 python：python3.10
apex==0.1+gitdb7007a.dtk2210 apex==1.1.0+das1.1.gitf477a3a.abi1.dtk2404.torch2.1.0
torch==1.10.0a0+git2040069.dtk2210 torch==2.1.0+das1.1.git3ac1bdd.abi1.dtk2404
``` ```
`Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应` `Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
......
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