In short
1 conda install -c conda-forge cudatoolkit-dev -y
1 2 which nvcc nvcc --version
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