Open remsky opened 2 months ago
If you get the "Torch not compiled with CUDA enabled" error, uninstall torch with:
pip uninstall torch
and then reinstall it with:
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124
then could solve the issues
Digs deep into torch and disables the pytorch cuda apparently, uninstalling and reinstalling everything as it goes.
If you've struggled with keeping your gpu libs balanced like I have, you'll be glad if you keep a backup env. Reinstalling everything again is a slow process
AssertionError: Torch not compiled with CUDA enabled**