Skip to content

GitLab

Commit c04a4bcf authored by Guolin Ke's avatar Guolin Ke
Browse files

fix docker build error

parent 8f0e59dc
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
docker/cu116/Dockerfile
View file @ c04a4bcf
...@@ -103,6 +103,8 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.5;8.0" ...@@ -103,6 +103,8 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.5;8.0"
RUN conda install -y numpy pyyaml scipy ipython mkl mkl-include ninja cython typing pandas && \ RUN conda install -y numpy pyyaml scipy ipython mkl mkl-include ninja cython typing pandas && \
conda clean -ya conda clean -ya
RUN pip3 install --upgrade sentry-sdk requests
RUN conda install pytorch=1.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge && \ RUN conda install pytorch=1.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge && \
conda clean -ya conda clean -ya
......
docker/rdma/Dockerfile
View file @ c04a4bcf
...@@ -68,6 +68,8 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.5;8.0" ...@@ -68,6 +68,8 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.5;8.0"
RUN conda install -y ninja typing && \ RUN conda install -y ninja typing && \
conda clean -ya conda clean -ya
RUN pip3 install --upgrade sentry-sdk requests
RUN pip3 install --no-cache-dir torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu116 && rm -rf ~/.cache/pip RUN pip3 install --no-cache-dir torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu116 && rm -rf ~/.cache/pip
RUN cd /tmp && \ RUN cd /tmp && \
......
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