Spaces:

KalbeDigitalLab
/

medsam-segment-anything

Running

haritsahm commited on Jul 21, 2023

Commit

40c1653

1 Parent(s): 3cfd2c8

increase ubuntu cuda

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.3.1-runtime-ubuntu20.04
 RUN apt update && \
     apt install -y bash \
@@ -6,7 +6,7 @@ RUN apt update && \
                    git \
                    curl \
                    ca-certificates \
-                   python3 \
                    python3-pip && \
     rm -rf /var/lib/apt/lists

+FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04
 RUN apt update && \
     apt install -y bash \
                    git \
                    curl \
                    ca-certificates \
+                   python3-dev \
                    python3-pip && \
     rm -rf /var/lib/apt/lists