Open jiayhuang opened 1 year ago
Maybe your pytorch version mismatch your cuda version?
This can happen when you have installed CUDA yourself at 12.1 and you are getting pytorch from pip. Pip is giving you a binary of pytorch which was built with CUDA 11.7. You need to downgrade CUDA to 11.7 or use the docker image which has CUDA 11.7.
I met some pip installation errors of
when I am following the instruction to install the project.
From some stackoverflow searches, looks like we need to compile Pytorch for CUDA 12 in order to fix it. Not quite sure if there is an easier way to fix the issue here :)