Open Zhuzi24 opened 2 years ago
Which version of cuda is needed, I reported AssertionError: Torch not compiled with CUDA enabled, I installed cuda11.1, it shows that it can be used .