apex not needed as amp is part of pytorch (#696)
Browse files- docker/Dockerfile-base +0 -5
docker/Dockerfile-base
CHANGED
|
@@ -57,11 +57,6 @@ FROM base-builder
|
|
| 57 |
ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 9.0+PTX"
|
| 58 |
ENV TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST
|
| 59 |
|
| 60 |
-
# recompile apex
|
| 61 |
-
RUN python3 -m pip uninstall -y apex
|
| 62 |
-
RUN git clone https://github.com/NVIDIA/apex
|
| 63 |
-
RUN cd apex && python3 -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./
|
| 64 |
-
|
| 65 |
RUN mkdir -p /workspace/builds
|
| 66 |
COPY --from=bnb-builder /workspace/bitsandbytes /workspace/builds/bitsandbytes
|
| 67 |
|
|
|
|
| 57 |
ARG TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 9.0+PTX"
|
| 58 |
ENV TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST
|
| 59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
RUN mkdir -p /workspace/builds
|
| 61 |
COPY --from=bnb-builder /workspace/bitsandbytes /workspace/builds/bitsandbytes
|
| 62 |
|