Skip to content

Commit

Permalink
remove xformers from image as github worker doesn't have enough capac…
Browse files Browse the repository at this point in the history
…ity to compile them from source
  • Loading branch information
ani300 committed Aug 9, 2023
1 parent 4c5dbae commit 60fbd99
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions nightly-pytorch-compile/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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} && \
Expand Down

0 comments on commit 60fbd99

Please sign in to comment.