Skip to content

Commit

Permalink
Reduce jobs to save ram for now
Browse files Browse the repository at this point in the history
  • Loading branch information
ani300 committed Aug 9, 2023
1 parent e94b0bc commit 4c5dbae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nightly-pytorch-compile/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ RUN pip install \

# xformers install (built from source)
ENV CUDA_HOME=/usr/local/cuda
ENV MAX_JOBS=2
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 && \
Expand Down

0 comments on commit 4c5dbae

Please sign in to comment.