Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +6 -0
Dockerfile
CHANGED
|
@@ -2,6 +2,12 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
|
| 2 |
|
| 3 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
# Set the MKL_THREADING_LAYER environment variable to GNU
|
| 6 |
ENV MKL_THREADING_LAYER=GNU
|
| 7 |
|
|
|
|
| 2 |
|
| 3 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 4 |
|
| 5 |
+
# Set CUDA_HOME environment variable
|
| 6 |
+
ENV CUDA_HOME=/usr/local/cuda
|
| 7 |
+
ENV PATH=${CUDA_HOME}/bin:/home/user/.local/bin:${PATH}
|
| 8 |
+
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
|
| 9 |
+
ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
|
| 10 |
+
|
| 11 |
# Set the MKL_THREADING_LAYER environment variable to GNU
|
| 12 |
ENV MKL_THREADING_LAYER=GNU
|
| 13 |
|