update kunlun dockerfile
This commit is contained in:
@@ -5,4 +5,6 @@ WORKDIR /workspace
|
||||
RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && pip install imageio[ffmpeg] einops datasets==3.1.0 simplejson addict sortedcontainers modelscope==1.28.2 av==11.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
|
||||
RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && pip install megatron megatron-core urllib3==1.26.20 -i https://pypi.tuna.tsinghua.edu.cn/simple
|
||||
RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && pip install open_clip_torch==2.24.0 pytorch-lightning==2.0.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
|
||||
|
||||
ENV PATH=/root/miniconda/envs/python38_torch201_cuda/bin:/root/miniconda/condabin:/opt/linux-bcecmd-0.3.3:/usr/local/go/bin:/bin:/opt/cmake-3.22.2/bin:/opt/just:/root/.cargo/bin:/usr/local/cuda-11.7/bin:/root/miniconda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/bin:/binnexport
|
||||
COPY . /workspace/
|
||||
|
||||
@@ -2,4 +2,5 @@
|
||||
# cnmon
|
||||
image=harbor-contest.4pd.io/zhanghao/t2v:r200-0.1
|
||||
device_id=2
|
||||
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --security-opt=seccomp=unconfined --cap-add=SYS_PTRACE --cap-add=SYS_ADMIN --device /dev/fuse --shm-size=32g --ulimit=memlock=-1 --ulimit=nofile=120000 --ulimit=stack=67108864 --device=/dev/xpu$device_id:/dev/xpu0 --device=/dev/xpuctrl:/dev/xpuctrl $image bash
|
||||
# docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --security-opt=seccomp=unconfined --cap-add=SYS_PTRACE --cap-add=SYS_ADMIN --device /dev/fuse --shm-size=32g --ulimit=memlock=-1 --ulimit=nofile=120000 --ulimit=stack=67108864 --device=/dev/xpu$device_id:/dev/xpu0 --device=/dev/xpuctrl:/dev/xpuctrl $image bash
|
||||
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=/dev/xpu$device_id:/dev/xpu0 --device=/dev/xpuctrl:/dev/xpuctrl $image bash
|
||||
|
||||
Reference in New Issue
Block a user