Annunziata Gianluca

Thank you! I gave a look but the models are for Tensorflow not Pytorch so are not compatible

To install torch with cuda look at this page https://pytorch.org/get-started/locally/ You need to select os, cuda version, and python package manager

You have to install python 3.8.10

Probably the problem is that the version 2.2 use pytorch-directml that use Directx12 not cuda. Try versions < 2.0 or just modify the code of 2.2 where the device =...

Ah sorry forgot to say, the AI models are not bundled on github because they are too heavy, you can download it here https://drive.google.com/drive/folders/13kfr3qny7S2xwG9h7v95F5mkWs0OmU0D __> BSRGAN.pth and RealSR_JPEG.pth

don t revert the entire commit, just modify 2.2 where there is "dml" string with cuda; so you will use all new feature came with 2.2 but with cuda backend

If you are using the script .py you can choose wich backend you want just installing the library and modifying the code. Yes, in the .zip there is only Pythorc-directml,...

strange, maibe there is something broken in pip packages installed. You can try to clean all packages installed and install everything again: 1. pip freeze > unistall.txt 2. pip uninstall...

You can also try Pytorch LTS with cuda 10: pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio===0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu102

You can use VsCode, it's much better