fix:add makefile and dockerfile
This commit is contained in:
47
Makefile
Normal file
47
Makefile
Normal file
@@ -0,0 +1,47 @@
|
||||
# Get version of CUDA and enable it for compilation if CUDA > 11.0
|
||||
# This solves https://github.com/IDEA-Research/Grounded-Segment-Anything/issues/53
|
||||
# and https://github.com/IDEA-Research/Grounded-Segment-Anything/issues/84
|
||||
# when running in Docker
|
||||
# Check if nvcc is installed
|
||||
NVCC := $(shell which nvcc)
|
||||
ifeq ($(NVCC),)
|
||||
# NVCC not found
|
||||
USE_CUDA := 0
|
||||
NVCC_VERSION := "not installed"
|
||||
else
|
||||
NVCC_VERSION := $(shell nvcc --version | grep -oP 'release \K[0-9.]+')
|
||||
USE_CUDA := $(shell echo "$(NVCC_VERSION) > 11" | bc -l)
|
||||
endif
|
||||
|
||||
# Add the list of supported ARCHs
|
||||
ifeq ($(USE_CUDA), 1)
|
||||
TORCH_CUDA_ARCH_LIST := "7.0;7.5;8.0;8.6+PTX"
|
||||
BUILD_MESSAGE := "I will try to build the image with CUDA support"
|
||||
else
|
||||
TORCH_CUDA_ARCH_LIST :=
|
||||
BUILD_MESSAGE := "CUDA $(NVCC_VERSION) is not supported"
|
||||
endif
|
||||
|
||||
|
||||
prepare-checkpoints:
|
||||
@echo "Preparing checkpoints..."
|
||||
cd checkpoints && \
|
||||
bash download_ckpts.sh && \
|
||||
mkdir -p gdino_checkpoints && \
|
||||
cd gdino_checkpoints && \
|
||||
wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth && \
|
||||
wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha2/groundingdino_swinb_cogcoor.pth
|
||||
|
||||
|
||||
build-image:
|
||||
@echo $(BUILD_MESSAGE)
|
||||
docker build --build-arg USE_CUDA=$(USE_CUDA) \
|
||||
--build-arg TORCH_ARCH=$(TORCH_CUDA_ARCH_LIST) \
|
||||
-t grounded_sam2:1.0 .
|
||||
run:
|
||||
docker run --gpus all -it --rm --net=host --privileged \
|
||||
-v /tmp/.X11-unix:/tmp/.X11-unix \
|
||||
-v "${PWD}":/home/appuser/Grounded-SAM-2 \
|
||||
-e DISPLAY=$DISPLAY \
|
||||
--name=gsa \
|
||||
--ipc=host -it grounded_sam2:1.0
|
Reference in New Issue
Block a user