Skip to content

GitLab

"src/targets/gpu/jit/ck_elementwise.cpp" did not exist on "4ec35e5fbe84a43c14bc9cfbcb0a9896d87cbde9"
Commit 677899c1 authored by wanglch's avatar wanglch
Browse files

Update Dockerfile

parent 2fffa2d6
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
docker/Dockerfile
View file @ 677899c1
...@@ -2,5 +2,3 @@ FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu22.04-dtk23.10. ...@@ -2,5 +2,3 @@ FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu22.04-dtk23.10.
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
COPY requirements.txt requirements.txt COPY requirements.txt requirements.txt
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
RUN pip install bitsandbytes-0.43.0-py3-none-any.whl
RUN pip install deepspeed-0.12.3+gitfe61783.abi0.dtk2310.torch2.1.0a0-cp310-cp310-manylinux2014_x86_64.whl
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