Spaces:

TRI-ML
/

vlm-demo

Paused

mattb512 commited on Apr 9, 2024

Commit

1e6d6d0

unverified ·

1 Parent(s): 3c2d5c1

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.3.2-base-ubuntu22.04
 RUN apt-get -y update \
     && apt-get -y upgrade \
@@ -27,4 +27,4 @@ RUN pip install -U --no-cache-dir .
 RUN pip list
-CMD sh $HOME/app/startup.sh

+FROM nvidia/cuda:12.4.0-base-ubuntu22.04
 RUN apt-get -y update \
     && apt-get -y upgrade \
 RUN pip list
+CMD sh $HOME/app/startup.sh