Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -26,8 +26,8 @@ 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 -
|
| 30 |
-
|
| 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
|
|
|
|
| 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 -xvf ${CUDNN_TAR_FILE}
|
| 30 |
+
RUN 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
|