UMS-Agenten/Base-Images
3
0
Fork 0
Code Actions Packages Releases Activity

Remove APEX
All checks were successful
Build and push Docker image in git push / build (push) Successful in 57s
Details

Browse Source
This commit is contained in:
Magnus Bender
2024-10-04 19:40:47 +02:00
parent fdbbbc3e3d
commit 5f54f79445
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.gpu
View File

@@ -23,10 +23,6 @@ RUN apt update && \
RUN python3 -m pip install --no-cache-dir --break-system-packages \ RUN python3 -m pip install --no-cache-dir --break-system-packages \
torch --index-url https://download.pytorch.org/whl/cu124 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 # fix library path for tensorflow and link a file
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/cuda/lib64" 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 RUN ln -s /usr/local/cuda/lib64/libcusolver.so.11 /usr/local/cuda/lib64/libcusolver.so.10