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 && \ ...@@ -53,7 +53,7 @@ RUN pip uninstall -y torch && \
pip uninstall -y torch && \ pip uninstall -y torch && \
conda clean -ya conda clean -ya
RUN conda install -y scipy scikit-learn pyyaml tensorboard tensorboardX && \ RUN conda install -y pyyaml tensorboardX && \
conda clean -ya conda clean -ya
# RUN ldconfig # RUN ldconfig
...@@ -63,7 +63,7 @@ RUN conda install -y scipy scikit-learn pyyaml tensorboard tensorboardX && \ ...@@ -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" 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 conda clean -ya
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 && \
......
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