Add LD_LIBRARY_PATH for pip-installed NVIDIA libs (bitsandbytes fix)
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -21,6 +21,7 @@ ENV CXX=/usr/bin/g++
|
|
| 21 |
RUN useradd -m -u 1000 user
|
| 22 |
USER user
|
| 23 |
ENV PATH="/home/user/.local/bin:$PATH"
|
|
|
|
| 24 |
|
| 25 |
WORKDIR /app
|
| 26 |
|
|
|
|
| 21 |
RUN useradd -m -u 1000 user
|
| 22 |
USER user
|
| 23 |
ENV PATH="/home/user/.local/bin:$PATH"
|
| 24 |
+
ENV LD_LIBRARY_PATH="/home/user/.local/lib/python3.10/site-packages/nvidia/nvjitlink/lib:/home/user/.local/lib/python3.10/site-packages/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH"
|
| 25 |
|
| 26 |
WORKDIR /app
|
| 27 |
|