AJMC2002 commited on
Commit
777c1c7
1 Parent(s): 1afe440

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -22,5 +22,10 @@ USER user
22
 
23
  COPY --chown=user public-apps/ /public-apps
24
  COPY --chown=user plates/ /home/user/plates
 
25
  RUN nvcc --version
 
 
 
 
26
  RUN /app/bin/warmup_apps
 
22
 
23
  COPY --chown=user public-apps/ /public-apps
24
  COPY --chown=user plates/ /home/user/plates
25
+
26
  RUN nvcc --version
27
+ RUN apt-cache policy libcudnn8 | head -n 3
28
+ RUN nvidia-smi
29
+ RUN apt-get install libcudnn8
30
+
31
  RUN /app/bin/warmup_apps