项目:
https://github.com/luca-medeiros/lang-segment-anything
环境:wsl ubuntu22.04
cuda12.1
sudo apt install g++
配合我之前juypter翻墙,重新运行
%pip install -e .
主要是之前在ubuntu24.04中不成功,现在成功记录如下:
cuda12.1
显卡4060 8g
系统:ubuntu24.04
显卡驱动:nvidia-driver-535(专有,测试)
之前一直报NameError: name ‘_C‘ is not defined未查看详细原因,后单独安装
GroundingDINO找到原因,gcc版本过高,不能超过12
这样弄:
sudo apt-get install gcc-12
sudo apt-get install g++-12
sudo ln -s /usr/bin/gcc-12 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-12 /usr/local/cuda/bin/g++
配合我之前juypter翻墙,重新运行
%pip install -e .
之后成功运行例子使用显卡加速
但只能运行一次,查看后发现显卡不会自动清缓存
google后,使用命令:
import torch
torch.cuda.empty_cache()
成功解决问题