NeuralNeighborStyleTransfer icon indicating copy to clipboard operation
NeuralNeighborStyleTransfer copied to clipboard

CPU mode not working without Nvidea Card?

Open krummrey opened this issue 2 years ago • 0 comments

I'm trying to install NNST on an ARM Ubuntu 22.04 virtual machine.

Seems like Torch doesn't work on that setup?

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

krummrey avatar Jun 27 '22 07:06 krummrey