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

fix docker build error

parent 8f0e59dc
......@@ -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 && \
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 && \
conda clean -ya
......
......@@ -68,6 +68,8 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.5;8.0"
RUN conda install -y ninja typing && \
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 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