You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
AssertionError: Torch not compiled with CUDA enabled
问题已解决:
pip install -r requirements.txt执行之后,安装的torch是cpu版本的,这导致后续运行时报错无法调用CUDA
解决方法是卸了该环境下的torch torchvision torchaudio去Pytorch官网重新下载