SAM2跑通（Ubuntu20.04)内含安装多个cuda

参考链接：

github链接
安装cuda，之前借鉴的方法
安装多个cuda

补充cuda安装：

A symlink already exists at /usr/local/cuda. Update to this installation?
选择no，否则会创建一个软连接覆盖之前那个/usr/local/cuda

sudo gedit ~/.bashrc
source ~/.bashrc
如果按第二个链接安装的cuda，手动改一下版本即可

SAM环境安装步骤

除了本地cuda版本要对应pytorch的cuda版本之外，没有什么难点。
但YOLO为啥不需要本地cuda，导致我一直以为不需要装cuda就能跑项目，pytorch自带？

git clone https://github.com/facebookresearch/segment-anything-2.git

cd segment-anything-2

conda create -n SAM python=3.10
conda activate SAM

# setup.py要求了环境，python>=3.10
pip install -e .
问题：RuntimeError:
The detected CUDA version (11.7) mismatches the version that was used to compile
PyTorch (12.1). Please make sure to use the same CUDA versions.
解决办法：先去pytorch官网查看torch>=2.3.1要求的cuda为11.8至少，只能再装个cuda12.1

pip install -e ".[demo]"

#下载权重
cd checkpoints
./download_ckpts.sh

SAM推理步骤

有点晕，感觉是标注软件的感觉，需要提供点或者框，然后进行预测。
这里展示使用点预测分割的效果，只需要修改image和input_point，其它详细信息可以查看github或者其他介绍。

import torch
from sam2.build_sam import build_sam2
from sam2.sam2_image_predictor import SAM2ImagePredictor
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

def show_mask(mask, ax, random_color=False, borders = True):
    if random_color:
        color = np.concatenate([np.random.random(3), np.array([0.6])], axis=0)
    else:
        color = np.array([30/255