AJMC2002 commited on
Commit
11a0e2d
·
verified ·
1 Parent(s): 676fbad

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- sudo cp -P cuda/include/cudnn.h /usr/local/cuda-12.6/include && \
32
- sudo cp -P cuda/lib/libcudnn* /usr/local/cuda-12.6/lib64/ && \
33
- sudo chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
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