项目:

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()

成功解决问题

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐