HorizonRobotics / Sparse4D

MIT License
326 stars 31 forks source link

运行bash local_test.sh出错 #27

Closed wofeifeileni closed 8 months ago

wofeifeileni commented 8 months ago

您好,运行验证命令时出现错误,似乎是提示CUDA不可用,我的环境里已经按照要求安装了torch==1.13.0+cu116,请问如何解决,十分感谢。 2024-03-21 00-15-29 的屏幕截图

linxuewu commented 8 months ago

local_test里面指定了CUDA_VISIBLE_DEVICES=3,看看有没有这张卡或者是不是这张卡被占满了。

wofeifeileni commented 8 months ago

只有一张卡,设置CUDA_VISIBLE_DEVICES=1,仍然出现以上问题

linxuewu commented 8 months ago

只有一张卡,CUDA_VISIBLE_DEVICES应该设为0。或者别指定

wofeifeileni commented 8 months ago

成功解决了,感谢