Yuliang commited on
Commit
0822a02
·
verified ·
1 Parent(s): 01cb6e2

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -97,4 +97,6 @@ COPY --chown=user . $HOME/app
97
  # Set the working directory to the user's home directory
98
  WORKDIR $HOME/app
99
 
 
 
100
  CMD ["python", "app.py"]
 
97
  # Set the working directory to the user's home directory
98
  WORKDIR $HOME/app
99
 
100
+ RUN which nvcc
101
+
102
  CMD ["python", "app.py"]