conda env config vars set LD_LIBRARY_PATH="/home/cyl/miniconda3/envs/dinov2/lib/" conda env config vars set CPATH="/home/cyl/miniconda3/envs/dinov2/include/" conda env config vars set CUDA_HOME="/home/cyl/miniconda3/envs/dinov2/"
export DATASET=/pth/to/dataset # path to your coco data
一些绊脚石 ^ ^
1
根据[[Cuda+Torch]],需要先安装cudatoolkit和cuda-toolkit
1 2 3 4 5
conda install nvidia/label/cuda-11.7.0::cuda-toolkit -c nvidia/label/cuda-11.7.0 conda install cudatoolkit # no need to specify version conda env config vars set LD_LIBRARY_PATH="/home/cyl/miniconda3/envs/<name>/lib/" conda env config vars set CPATH="/home/cyl/miniconda3/envs/<name>/include/" conda env config vars set CUDA_HOME="/home/cyl/miniconda3/envs/<name>/"
如果编译时出现ld: cannot find -lcudart: No such file or directory collect2: error: ld returned 1 exit status 报错,只是因为没有安装cudatoolkit ^ ^
3
安装完成后直接import semantic_sam会报错ModuleNotFoundError: No module named 'MultiScaleDeformableAttention' ^ ^ 提示:
1 2 3
Please compile MultiScaleDeformableAttention CUDA op with the following commands: `cd mask2former[/modeling/pixel_decoder/ops](http://127.0.0.1:8888/modeling/pixel_decoder/ops)` `sh make.sh`
需要手动make一下 Mask2Former:
1 2
cd Mask2Former/mask2former/modeling/pixel_decoder/ops/ sh make.sh