From 60fbd999084b64d1c4e52bc9efbef90691ab0f67 Mon Sep 17 00:00:00 2001 From: Antoni Viros i Martin Date: Wed, 9 Aug 2023 15:11:52 -0400 Subject: [PATCH] remove xformers from image as github worker doesn't have enough capacity to compile them from source --- nightly-pytorch-compile/Dockerfile | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/nightly-pytorch-compile/Dockerfile b/nightly-pytorch-compile/Dockerfile index a609b84..9641d11 100644 --- a/nightly-pytorch-compile/Dockerfile +++ b/nightly-pytorch-compile/Dockerfile @@ -48,15 +48,16 @@ RUN pip install \ chmod -R g=u /opt/conda # xformers install (built from source) -ENV CUDA_HOME=/usr/local/cuda -ENV MAX_JOBS=1 -ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6" -RUN apt-get install -y libcublas-11-8 libcublas-dev-11-8 libcusparse-11-8 libcusparse-dev-11-8 libcusolver-11-8 libcusolver-dev-11-8 libcurand-11-8 libcurand-dev-11-8 && \ - pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers --no-cache-dir && \ - chmod -R g+w /opt/conda && \ - chmod -R g=u /opt/conda && \ - apt-get remove -y libcublas-11-8 libcublas-dev-11-8 libcusparse-11-8 libcusparse-dev-11-8 libcusolver-11-8 libcusolver-dev-11-8 libcurand-11-8 libcurand-dev-11-8 && \ - apt-get -y clean +# deactivated until we get better workers as there isn't enough ram, cpu, or disk space to compile them anymore +# ENV CUDA_HOME=/usr/local/cuda +# ENV MAX_JOBS=2 +# ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6" +# RUN apt-get install -y libcublas-11-8 libcublas-dev-11-8 libcusparse-11-8 libcusparse-dev-11-8 libcusolver-11-8 libcusolver-dev-11-8 libcurand-11-8 libcurand-dev-11-8 && \ +# pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers --no-cache-dir && \ +# chmod -R g+w /opt/conda && \ +# chmod -R g=u /opt/conda && \ +# apt-get remove -y libcublas-11-8 libcublas-dev-11-8 libcusparse-11-8 libcusparse-dev-11-8 libcusolver-11-8 libcusolver-dev-11-8 libcurand-11-8 libcurand-dev-11-8 && \ +# apt-get -y clean # setup .aim_profile (needed to get aim running) RUN mkdir -p ${HOME} && \