Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM nvidia/cuda:12.
|
| 2 |
|
| 3 |
RUN apt-get -y update \
|
| 4 |
&& apt-get -y upgrade \
|
|
@@ -27,4 +27,4 @@ RUN pip install -U --no-cache-dir .
|
|
| 27 |
|
| 28 |
RUN pip list
|
| 29 |
|
| 30 |
-
CMD sh $HOME/app/startup.sh
|
|
|
|
| 1 |
+
FROM nvidia/cuda:12.4.0-base-ubuntu22.04
|
| 2 |
|
| 3 |
RUN apt-get -y update \
|
| 4 |
&& apt-get -y upgrade \
|
|
|
|
| 27 |
|
| 28 |
RUN pip list
|
| 29 |
|
| 30 |
+
CMD sh $HOME/app/startup.sh
|