search

Jumpat / SegmentAnythingin3D

Segment Anything in 3D with NeRFs (NeurIPS 2023)
Apache License 2.0
831 stars 52 forks source link

cudnn error #74

Open zw20130717 opened 1 month ago

zw20130717 commented 1 month ago

i was running the command 'python run_seg_gui.py --config=configs/llff/seg/seg_fern.py --segment \ --sp_name=_gui --num_prompts=20 \ --render_opt=train --save_ckpt' when I received this cudnn error information. I have already installed the corresponding cudnn version to my cuda version(11.3) and still cant solve this problem. I'd appreciate it if anyone can offer me some advice on it. 2024-06-04 15-48-34 的屏幕截图

zw20130717 commented 1 month ago

unable to find a valid cudnn algorithm to run convolution😢

Jumpat commented 1 month ago

Hi, have you checked 'nvcc -V' ? The 'CUDA_HOME' and 'LD_LIBRARY_PATH' ?

I think maybe torch cannot find the cuda. Does 'torch.cuda.is_available()' ?

zw20130717 commented 1 month ago

Hi, have you checked 'nvcc -V' ? The 'CUDA_HOME' and 'LD_LIBRARY_PATH' ?

I think maybe torch cannot find the cuda. Does 'torch.cuda.is_available()' ?

yeah I THINK torch.cuda.is_available is true and cuda_home and ld_library_path is set. 2024-06-04 17-34-30 的屏幕截图

Jumpat commented 1 month ago

Hi, you may check this for some help. I have never met such problem in sa3d.

zw20130717 commented 1 month ago

Hi, you may check this for some help. I have never met such problem in sa3d.

Hi, I think I have solved the CUDNN problem but just received cuda-out-of-memory. May I ask your computer configuration to run this project? Mine is GEFORCE RTX 3070 with 8GB memory