SegmentAnythingin3D icon indicating copy to clipboard operation
SegmentAnythingin3D copied to clipboard

cudnn error

Open zw20130717 opened this issue 1 year ago • 5 comments

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 avatar Jun 04 '24 07:06 zw20130717

unable to find a valid cudnn algorithm to run convolution😢

zw20130717 avatar Jun 04 '24 08:06 zw20130717

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

Jumpat avatar Jun 04 '24 09:06 Jumpat

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 的屏幕截图

zw20130717 avatar Jun 05 '24 02:06 zw20130717

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

Jumpat avatar Jun 05 '24 02:06 Jumpat

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

zw20130717 avatar Jun 05 '24 03:06 zw20130717