Kornia and PyTorch dependency issue in Ubuntu

[demiahmed]

Hi I tried installing the necessary dependencies using environment.yml and it throws me the error discussed here (kornia/kornia#1290)

I tried applying the fix which was to install Kornia from source using the command pip install git+https://github.com/kornia/kornia

However it ended up installing torch==1.9.1 as a dependency (environment.yml in this repo has a torch==1.4) and breaks the entire dependency tree.

Is there a fix to get the most recent version of Kornia that's compatible with this code and with torch==1.4.0 or any possible workarounds?