lingyuzeng/nvidia_docker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Your Name
2024-08-02 16:14:01 +08:00
parent 8a041dccd7
commit 7b48198f1d
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile.colossalai
View File

@@ -43,4 +43,4 @@ RUN git clone -b ${VERSION} https://github.com/hpcaitech/ColossalAI.git \
# install tensornvme # install tensornvme
RUN conda install -y cmake && \ RUN conda install -y cmake && \
apt update -y && apt install -y libaio-dev && \ apt update -y && apt install -y libaio-dev && \
pip install -v git+https://github.com/hpcaitech/TensorNVMe.git pip install -v git+https://github.com/hpcaitech/TensorNVMe.git

4
docker-compose_ngc.yml
View File

@@ -30,8 +30,8 @@ services:
DS_BUILD_OPS: 1 DS_BUILD_OPS: 1
DEEPSPEED_VERSION: "master" DEEPSPEED_VERSION: "master"
DEEPSPEED_INSTALL_FLAGS: "--allow_sudo" DEEPSPEED_INSTALL_FLAGS: "--allow_sudo"
HTTP_PROXY: "http://127.0.0.1:15777" # HTTP_PROXY: "http://127.0.0.1:15777"
HTTPS_PROXY: "http://127.0.0.1:15777" # HTTPS_PROXY: "http://127.0.0.1:15777"
CACHEBUST: 1 CACHEBUST: 1
# volumes: # volumes:
# - ./workspace:/workspace # - ./workspace:/workspace