|
|
|
|
|
|
|
|
|
|
|
NVCC := $(shell which nvcc) |
|
ifeq ($(NVCC),) |
|
|
|
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 |
|
|
|
|
|
ifeq ($(USE_CUDA), 1) |
|
TORCH_CUDA_ARCH_LIST := "3.5;5.0;6.0;6.1;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 |
|
|
|
|
|
build-image: |
|
@echo $(BUILD_MESSAGE) |
|
docker build --build-arg USE_CUDA=$(USE_CUDA) \ |
|
--build-arg TORCH_ARCH=$(TORCH_CUDA_ARCH_LIST) \ |
|
-t gsa:v0 . |
|
run: |
|
ifeq (,$(wildcard ./sam_vit_h_4b8939.pth)) |
|
wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth |
|
endif |
|
ifeq (,$(wildcard ./groundingdino_swint_ogc.pth)) |
|
wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth |
|
endif |
|
docker run --gpus all -it --rm --net=host --privileged \ |
|
-v /tmp/.X11-unix:/tmp/.X11-unix \ |
|
-v "${PWD}":/home/appuser/working_dir \ |
|
-e DISPLAY=$DISPLAY \ |
|
--name=gsa \ |
|
--ipc=host -it gsa:v0 |
|
|
