File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 852, in to
return self._apply(convert)
File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 530, in _apply
module._apply(fn)
File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 552, in _apply
param_applied = fn(param)
File "/usr/local/lib/python3.9/site-packages/torch/nn/modules/module.py", line 850, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File "/usr/local/lib/python3.9/site-packages/torch/cuda/__init__.py", line 166, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
Currently fails with no GPU: