linux服务器配置GroundingDINO 详细过程
1. 参考帖子
已经跑通了,该踩的坑也都踩过来了,大家按照我配置的流程来配,大概率不会出错。(现在是2024/5/31日)
Grounding DINO使用攻略一
Github- IDEA-Research / GroundingDINO Public
【AI】Windows环境安装GroundingDINO
Ground DINO 自用/部署方法(notebook代码)
大家主要参考相关的Github网站
2. 配置流程:
环境配置:py310, cuda118, pytorch2.1
2.1 设置相关的环境变量:
# 首先: 设置相关的CUDA_HOME
echo $CUDA_HOME
# 寻找一下我们这台电脑的nvcc
which nvcc
# 显示/usr/local/cuda/bin/nvcc
# 设置相关的环境变量
export CUDA_HOME=/usr/local/cuda
# 再次输入:
echo $CUDA_HOME
显示:/usr/local/cuda
# 如果想永久的设置好cuda home(建议永久设置),可以这样设置:
echo 'export CUDA_HOME=/usr/local/cuda' >> ~/.bashrc
source ~/.bashrc
echo $CUDA_HOME
2.2 配置conda
下载anaconda
首先进入清华大学开源镜像站,
点击里面的archive页面,下载Anaconda3-2024.02-1-Linux-x86_64.sh 安装包。 (但是夜里好像打不开清华大学开源镜像站,可能白天才能打开,可以换一个网址来下载。)
# 我使用的是autodl云服务器来跑的,这个服务器基本都把文件放到autodl-tmp文件夹下,大家进入自己的文件夹里就好
cd autodl-tmp
# 为Anaconda3-2024.02-1-Linux-x86_64.sh赋予运行权限
chmod +x Anaconda3-2024.02-1-Linux-x86_64.sh
# 开始安装
./Anaconda3-2024.02-1-Linux-x86_64.sh
# 之后在系统root根目录下运行
ls -a
# 之后可以看到相关的目录下有.bashrc文件,运行命令:
vim .bashrc
# 之后我们添加上一句: (因为我的conda装完以后默认就在/root/anaconda3下)
export PATH=/root/anaconda3/bin:$PATH
# 之后保存文件,然后
source ~/.bashrc
# 这样的话,我们默认运行的anaconda就是咱们自己安装的anaconda了。(系统里有两个conda,我制定了我自己安装的这个版本的conda,便于后续迁移到其他环境里)
# 我们运行命令:
conda info
# 只要我们包上传的过程不出错,安装的过程也不出错,同时全程正确的安装好了之后,这时显示的就是正确的conda信息、
配置相对应的环境
1. 先使用anadonda创建python3.10的DINO环境:
# 配置相对应的conda环境
source activate base
conda create -n dino python=3.10
conda init bash && source /root/.bashrc
conda activate dino
为了避免cuda版本错误,我在这里又额外利用conda安装了一下cuda: (服务器里已经有cuda11.8和cudnn的大家就不用再安装了)
conda search cudatoolkit
conda search cudnn
# 也可使用conda search cudatoolkit --info命令,下载后conda install ./cudatoolkit-11.3.1-h2bc3f7f_2.conda 本地安装
# 如果不怕慢,也可以换源搜索:
# conda search cudatoolkit -c conda-forge
# conda search cudnn -c conda-forge
conda install cudatoolkit==11.8.0
conda install cudnn==8.9.7.29
# 我这里本地下载安装了
# conda search cudatoolkit --info -c conda-forge
# conda install ./cudnn-8.9.7.29-hcdd5f01_2.conda
# 接下来可以检查一下conda安装的cuda和cudnn
conda list | grep cudatoolkit
conda list | grep cudnn
现在安装一下对应cuda11.8版本的pytorch2.1
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia
然后进入安装GroundingDINO:
# 首先获取相关的DINO的安装包
git clone https://github.com/IDEA-Research/GroundingDINO.git
cd GroundingDINO/
# 其次安装相关的DINO
pip install -e .
# 这回我