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
.github/workflows/base.yml vendored
View File

@@ -16,11 +16,11 @@ jobs:
include: include:
- cuda: cu118 - cuda: cu118
cuda_version: 11.8.0 cuda_version: 11.8.0
cuda_version_bnb: 118 cuda_version_bnb: "118"
pytorch: 2.0.0 pytorch: 2.0.0
- cuda: cu117 - cuda: cu117
cuda_version: 11.7.0 cuda_version: 11.7.0
cuda_version_bnb: 117 cuda_version_bnb: "117"
pytorch: 1.13.1 pytorch: 1.13.1
steps: steps:
- name: Checkout - name: Checkout

4
docker/Dockerfile-base
View File

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