diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu index 3b25a2c..b3f35a5 100644 --- a/docker/Dockerfile.gpu +++ b/docker/Dockerfile.gpu @@ -23,10 +23,6 @@ RUN apt update && \ RUN python3 -m pip install --no-cache-dir --break-system-packages \ torch --index-url https://download.pytorch.org/whl/cu124 -RUN git clone https://github.com/NVIDIA/apex -RUN cd apex && \ - pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --break-system-packages --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./ - # fix library path for tensorflow and link a file ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/cuda/lib64" RUN ln -s /usr/local/cuda/lib64/libcusolver.so.11 /usr/local/cuda/lib64/libcusolver.so.10