In short

1
conda install -c conda-forge cudatoolkit-dev -y
1
2
which nvcc
nvcc --version
1
echo $CUDA_HOME
1
2
export CUDA_HOME=$CONDA_PREFIX
echo $CUDA_HOME

GroundingDINO and Segment-Anything

I was trying to install Grounded-Segment-Anything but somehow I cannot run it in GPUmode.
After a few hours of searching, I realized the package I installed was CPUmode-only.

Therefore the full step to install Grounded-Segment-Anything:

1
2
3
4
5
6
7
conda install -c conda-forge cudatoolkit-dev -y
export BUILD_WITH_CUDA=True
export CUDA_HOME=$CONDA_PREFIX

git clone https://github.com/IDEA-Research/Grounded-Segment-Anything.git && cd Grounded-Segment-Anything
python -m pip install -e segment_anything
python -m pip install -e GroundingDINO