Skip to content

GitLab

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

remove duplicate packages

parent 76fee8b6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/rdma/Dockerfile
View file @ a95f080b
......@@ -53,7 +53,7 @@ RUN pip uninstall -y torch && \
pip uninstall -y torch && \
conda clean -ya
RUN conda install -y scipy scikit-learn pyyaml tensorboard tensorboardX && \
RUN conda install -y pyyaml tensorboardX && \
conda clean -ya
# RUN ldconfig
......@@ -63,7 +63,7 @@ RUN conda install -y scipy scikit-learn pyyaml tensorboard tensorboardX && \
# # ------------------------------------------------------------------
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 ninja typing && \
conda clean -ya
RUN conda install pytorch=1.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge && \
......
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