AJMC2002 commited on
Commit
9844e73
·
verified ·
1 Parent(s): cd35ff0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -26,9 +26,9 @@ COPY --chown=user plates/ /home/user/plates
26
  USER root
27
  RUN 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
28
  RUN tar -xvf cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz
29
- RUN sudo cp -P cuda/include/cudnn.h /usr/local/cuda-12.6/include
30
- RUN sudo cp -P cuda/lib/libcudnn* /usr/local/cuda-12.6/lib64/
31
- RUN sudo chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
32
  USER user
33
 
34
  RUN /app/bin/warmup_apps
 
26
  USER root
27
  RUN 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
28
  RUN tar -xvf cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz
29
+ RUN cp -P cudnn-linux-x86_64-9.1.0.70_cuda12-archive/include/cudnn.h /usr/local/cuda-12.6/include
30
+ RUN cp -P cudnn-linux-x86_64-9.1.0.70_cuda12-archive/lib/libcudnn* /usr/local/cuda-12.6/lib64/
31
+ RUN chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
32
  USER user
33
 
34
  RUN /app/bin/warmup_apps