Update Dockerfile

supervisely-ecosystem · Oct 8, 2024 · 25ab34c · 25ab34c
1 parent dc39301
commit 25ab34c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,6 +1,6 @@
 FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
-RUN apt update && apt install python3-pip wget -y
+RUN apt update && apt install python3-pip wget curl -y
 RUN apt-get install -y git
 
 RUN pip3 install torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121
@@ -26,9 +26,9 @@ RUN pip3 install mock==5.1.0
 RUN mkdir -p sam2.1_weights
 
 # Download SAM 2.1 checkpoints
-RUN wget -O /sam2_weights/sam2.1_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt"
-RUN wget -O /sam2_weights/sam2.1_hiera_small.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt"
-RUN wget -O /sam2_weights/sam2.1_hiera_base_plus.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt"
-RUN wget -O /sam2_weights/sam2.1_hiera_large.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt"
+RUN curl -L -o /sam2_weights/sam2.1_hiera_tiny.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt"
+RUN curl -L -o /sam2_weights/sam2.1_hiera_small.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt"
+RUN curl -L -o /sam2_weights/sam2.1_hiera_base_plus.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt"
+RUN curl -L -o /sam2_weights/sam2.1_hiera_large.pt "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt"
 
 LABEL python_sdk_version=6.73.203