AJMC2002 commited on
Commit
54e9b2d
·
verified ·
1 Parent(s): 0e911cd

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -2
Dockerfile CHANGED
@@ -23,12 +23,15 @@ USER user
23
  COPY --chown=user public-apps/ /public-apps
24
  COPY --chown=user plates/ /home/user/plates
25
 
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
 
23
  COPY --chown=user public-apps/ /public-apps
24
  COPY --chown=user plates/ /home/user/plates
25
 
26
+ USER root
27
+ RUN apt-get install libcudnn8
28
+ RUN apt-get install libcudnn8-dev
29
+ RUN apt-get install libcudnn8-samples
30
  #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
31
  #RUN tar -xvf cudnn-linux-x86_64-9.1.0.70_cuda12-archive.tar.xz
32
  #RUN cp -P cudnn-linux-x86_64-9.1.0.70_cuda12-archive/include/cudnn.h /usr/local/cuda-12.6/include
33
  #RUN cp -P cudnn-linux-x86_64-9.1.0.70_cuda12-archive/lib/libcudnn* /usr/local/cuda-12.6/lib64/
34
  #RUN chmod a+r /usr/local/cuda-12.6/lib64/libcudnn*
35
+ USER user
36
 
37
  RUN /app/bin/warmup_apps