Skip to content

GitLab

Commit 07f95057 authored by Guolin Ke's avatar Guolin Ke
Browse files

use pip to install torch

parent a95f080b
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 8 deletions
docker/rdma/Dockerfile
View file @ 07f95057
...@@ -49,8 +49,9 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ ...@@ -49,8 +49,9 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
debhelper \ debhelper \
nfs-common nfs-common
RUN pip uninstall -y torch && \ RUN pip uninstall -y torch torchvision torchtext && \
pip uninstall -y torch && \ pip uninstall -y torch torchvision torchtext && \
rm -rf ~/.cache/pip && \
conda clean -ya conda clean -ya
RUN conda install -y pyyaml tensorboardX && \ RUN conda install -y pyyaml tensorboardX && \
...@@ -66,17 +67,15 @@ ENV TORCH_CUDA_ARCH_LIST "7.0;7.5;8.0" ...@@ -66,17 +67,15 @@ 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 conda install pytorch=1.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge && \ RUN pip3 install --no-cache-dir torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu116 && rm -rf ~/.cache/pip
conda clean -ya
RUN cd /tmp && \ RUN cd /tmp && \
git clone https://github.com/dptech-corp/Uni-Core && \ git clone https://github.com/dptech-corp/Uni-Core && \
cd Uni-Core && \ cd Uni-Core && \
python setup.py install &&\ python setup.py install && \
rm -rf /tmp/* rm -rf /tmp/* && rm -rf ~/.cache/pip
RUN pip install --no-cache-dir tokenizers lmdb biopython ml-collections timeout-decorator urllib3 tree dm-tree
RUN pip3 install --no-cache-dir tokenizers lmdb biopython ml-collections timeout-decorator urllib3 tree dm-tree && rm -rf ~/.cache/pip
RUN ldconfig && \ RUN ldconfig && \
apt-get clean && \ apt-get clean && \
......
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