Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +4 -5
Dockerfile
CHANGED
|
@@ -26,11 +26,10 @@ COPY --chown=user plates/ /home/user/plates
|
|
| 26 |
USER root
|
| 27 |
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 |
-
cp -P cuda/
|
| 32 |
-
|
| 33 |
-
chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
|
| 34 |
USER user
|
| 35 |
|
| 36 |
RUN /app/bin/warmup_apps
|
|
|
|
| 26 |
USER root
|
| 27 |
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 |
+
sudo cp -P cuda/include/cudnn.h /usr/local/cuda-12.6/include && \
|
| 31 |
+
sudo cp -P cuda/lib/libcudnn* /usr/local/cuda-12.6/lib64/ && \
|
| 32 |
+
sudo chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
|
|
|
|
| 33 |
USER user
|
| 34 |
|
| 35 |
RUN /app/bin/warmup_apps
|