| 12345678910111213141516 |
- # Base image must at least have pytorch and CUDA installed.
- ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:23.03-py3
- FROM $BASE_IMAGE
- ARG BASE_IMAGE
- RUN echo "Installing Apex on top of ${BASE_IMAGE}"
- # make sure we don't overwrite some existing directory called "apex"
- WORKDIR /tmp/unique_for_apex
- # uninstall Apex if present, twice to make absolutely sure :)
- RUN pip uninstall -y apex || :
- RUN pip uninstall -y apex || :
- # SHA is something the user can touch to force recreation of this Docker layer,
- # and therefore force cloning of the latest version of Apex
- RUN SHA=ToUcHMe git clone https://github.com/NVIDIA/apex.git
- WORKDIR /tmp/unique_for_apex/apex
- RUN pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .
- WORKDIR /workspace
|
