I'm trying to install NNST on an ARM Ubuntu 22.04 virtual machine.
python styleTransfer.py --content_path inputs/content/C1.png --style_path inputs/style/S4.jpg --output_path ./output.jpg --cpu
/home/jan/nnst/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
Traceback (most recent call last):
File "/home/jan/Dokumente/NeuralNeighborStyleTransfer/styleTransfer.py", line 63, in <module>
torch.cuda.synchronize()
File "/home/jan/nnst/lib/python3.9/site-packages/torch/cuda/__init__.py", line 493, in synchronize
_lazy_init()
File "/home/jan/nnst/lib/python3.9/site-packages/torch/cuda/__init__.py", line 210, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
I'm trying to install NNST on an ARM Ubuntu 22.04 virtual machine.
Seems like Torch doesn't work on that setup?