JiehongLin / SAM-6D

[CVPR2024] Code for "SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation".
353 stars 30 forks source link

some problem in prepare.sh #64

Open 3114917790 opened 2 months ago

3114917790 commented 2 months ago

there are some problem when i use python setup.py install [7/9] /usr/local/cuda/bin/nvcc -I_ext_src/include -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include/TH -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include/THC -I/usr/local/cuda/include -I/home/d1//environment/envs/sam6d/include/python3.9 -c -c /home/d1//SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/_ext_src/src/ball_query_gpu.cu -o /home/d1//SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/build/temp.linux-x86_64-cpython-39/_ext_src/src/ball_query_gpu.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -O3 -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_ext -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 -std=c++17 FAILED: /home/d1/zwb/SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/build/temp.linux-x86_64-cpython-39/_ext_src/src/ball_query_gpu.o /usr/local/cuda/bin/nvcc -I_ext_src/include -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include/torch/csrc/api/include -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include/TH -I/home/d1//environment/envs/sam6d/lib/python3.9/site-packages/torch/include/THC -I/usr/local/cuda/include -I/home/d1//environment/envs/sam6d/include/python3.9 -c -c /home/d1/zwb/SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/_ext_src/src/ball_query_gpu.cu -o /home/d1/zwb/SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/build/temp.linux-x86_64-cpython-39/_ext_src/src/ball_query_gpu.o -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_BFLOAT16_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -O3 -DCUDA_HAS_FP16=1 -DCUDA_NO_HALF_OPERATORS -DCUDA_NO_HALF_CONVERSIONS -DCUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=_ext -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 -std=c++17 /home/d1//SAM-6D/SAM-6D/Pose_Estimation_Model/model/pointnet2/_ext_src/src/ball_query_gpu.cu:10:10: fatal error: cuda_utils.h: No such file or directory

include "cuda_utils.h"

      ^~~~~~~~~~~~~~

compilation terminated.