Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -28,9 +28,9 @@ RUN CUDNN_TAR_FILE="cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz" && \
|
|
| 28 |
wget wget https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz && \
|
| 29 |
tar -xzvf ${CUDNN_TAR_FILE}
|
| 30 |
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
USER user
|
| 35 |
|
| 36 |
RUN /app/bin/warmup_apps
|
|
|
|
| 28 |
wget wget https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz && \
|
| 29 |
tar -xzvf ${CUDNN_TAR_FILE}
|
| 30 |
|
| 31 |
+
cp -P cuda/include/cudnn.h /usr/local/cuda-12.6/include && \
|
| 32 |
+
cp -P cuda/lib/libcudnn* /usr/local/cuda-12.6/lib64/ && \
|
| 33 |
+
chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
|
| 34 |
USER user
|
| 35 |
|
| 36 |
RUN /app/bin/warmup_apps
|