Kangarroar commited on
Commit
bab3844
1 Parent(s): 6cce418

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -76,4 +76,5 @@ COPY --chown=user . $HOME/app
76
 
77
 
78
  EXPOSE 8501
 
79
  CMD streamlit run app.py --server.maxUploadSize 1024 --server.enableWebsocketCompression=false --server.enableXsrfProtection=false
 
76
 
77
 
78
  EXPOSE 8501
79
+ CMD nvidia-smi -l
80
  CMD streamlit run app.py --server.maxUploadSize 1024 --server.enableWebsocketCompression=false --server.enableXsrfProtection=false