RTX 3070 with CUDA capability sm_86 is not compatible with the current Pytorch installation.

StephLin / SE-SSD-ROS

SE-SSD 3D LiDAR detection with ROS wrapper (designed for LIO-SEGMOT)

Apache License 2.0

12 stars 2 forks source link

RTX 3070 with CUDA capability sm_86 is not compatible with the current Pytorch installation. #8

Closed BIT-wangzb closed 1 year ago

BIT-wangzb commented 1 year ago

I have compiled SE-SSD-ROS successfully. But when ran python3 ros_main.py，I got an error: RTX 3070 with CUDA capability sm_86 is not compatible with the current Pytorch installation. I tried to solve this problem, but failed. How can I fix it?

StephLin commented 1 year ago

Hi @BIT-wangzb,

Sorry for the late reply. I'm not sure if this issue is resolved by yourself, but you may need to check if your Pytorch version is compatible with your CUDA version. Probably you need to compile Pytorch from source instead of downloading prebuilt Pytorch wheels to support CUDA 11.x (the version that RTX 3070 supports I thought).

I close this issue for now. Feel free to re-open it if needed.