LCORE-569: Fix installing cpu version of pytorch, drop LD_LIBRARY_PATH for gpu image (#34)

matysek · web-flow · commit 53179ec60ee4 · 2025-08-19T16:52:15.000+02:00
* LCORE-569: Fix installing the right pytorch cpu version

* LCORE-569: For gpu image drop LD_LIBRARY_PATH ooverride
diff --git a/Containerfile-gpu b/Containerfile-gpu
@@ -6,7 +6,6 @@ RUN dnf install -y --nodocs --setopt=keepcache=0 --setopt=tsflags=nodocs \
     python3.12 python3.12-devel python3.12-pip libcudnn9 libnccl libcusparselt0 && \
     dnf clean all
 RUN ln -sf /usr/bin/python3.12 /usr/bin/python
-ENV LD_LIBRARY_PATH=/usr/local/cuda-12/compat:$LD_LIBRARY_PATH
 
 # Install asciidoctor
 RUN dnf install -y rubygems && \
diff --git a/pyproject.toml b/pyproject.toml
@@ -58,7 +58,7 @@ name = "pytorch-cpu"
 url = "https://download.pytorch.org/whl/cpu"
 explicit = true
 [tool.uv.sources]
-torchvision = [{ index = "pytorch-cpu" }]
+torch = [{ index = "pytorch-cpu" }]
 
 [dependency-groups]
 dev = [
diff --git a/uv.lock b/uv.lock
