It is not running on GPU in google colab :/
See original GitHub issueHi, awesome repo. Thanks for sharing. It throws this error every time I try to run this on GPU.
Traceback (most recent call last): File "/usr/local/bin/style_transfer", line 33, in <module> sys.exit(load_entry_point('style-transfer-pytorch', 'console_scripts', 'style_transfer')()) File "/content/style-transfer-pytorch/style_transfer/cli.py", line 213, in main props = torch.cuda.get_device_properties(device) File "/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py", line 296, in get_device_properties _lazy_init() # will define _get_device_properties File "/usr/local/lib/python3.6/dist-packages/torch/cuda/__init__.py", line 172, in _lazy_init torch._C._cuda_init() RuntimeError: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.
So I’m forced to run it on the CPU instead and which takes forever. 😪 Can you make it compatible with the old drivers? 😅
I made this [colab] notebook. You can reproduce the error yourself.
Issue Analytics
- State:
- Created 3 years ago
- Comments:10 (5 by maintainers)
You can use multiple style images (just list the additional ones after the first one) and it will create a blended style from them and apply it to the content. You can also use the style weights parameter (
-sw
) to specify the ratios of the style images: for instance, if you have three style images,-sw 1 2 1
will use them in a 1:2:1 ratio.The values of the parameters are supposed to be independent of what size the output image is. But sometimes when producing large images I set the minimum scale (
-ms
) to 256 or thereabouts.