tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

move CUDA_VERSION_BNB arg inside of stage build scope

Browse Source
This commit is contained in:
Wing Lian
2023-05-29 13:30:15 -04:00
parent 00323f0a6f
commit e43bcc6c4f
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile-base
View File

@@ -1,5 +1,4 @@
ARG CUDA_VERSION="11.8.0"
ARG CUDA_VERSION_BNB="118"
ARG CUDNN_VERSION="8"
ARG UBUNTU_VERSION="22.04"
ARG MAX_JOBS=4
@@ -62,6 +61,7 @@ RUN git clone https://github.com/microsoft/DeepSpeed.git && \
FROM base-builder AS bnb-builder
WORKDIR /workspace
ARG CUDA_VERSION_BNB="118"
ENV CUDA_VERSION_BNB=$CUDA_VERSION_BNB
RUN git clone https://github.com/TimDettmers/bitsandbytes.git && \
@@ -71,8 +71,6 @@ RUN git clone https://github.com/TimDettmers/bitsandbytes.git && \
FROM base-builder
ENV CUDA_VERSION_BNB=$CUDA_VERSION_BNB
# recompile apex
RUN python3 -m pip uninstall -y apex
RUN git clone https://github.com/NVIDIA/apex