PKU-EPIC / GraspNeRF

[ICRA 2023] GraspNeRF: Multiview-based 6-DoF Grasp Detection for Transparent and Specular Objects Using Generalizable NeRF
129 stars 16 forks source link

Torch error during training and testing #23

Closed camilletaglione closed 8 months ago

camilletaglione commented 8 months ago

Hello, A really impressive contribution. I am running the code on an RTX A4000 I use Torch 2.1 with cuda 12.1 after launching both

bash run_simgrasp.sh

or

bash train.sh GPU_ID

I get the following error

File "/home/user/miniconda3/envs/GRASP/lib/python3.9/site-packages/torch/init.py", line 235, in from torch._C import * # noqa: F403 ImportError: /home/user/miniconda3/envs/GRASP/lib/python3.9/site-packages/torch/lib/libtorch_cpu.so: undefined symbol: cblas_gemm_bf16bf16f32

Do you have an idea, what can I do to fix that ? Thanks in advance