官方连接
kornia-GitHub

可行版本/GPU-3080

Ubuntu 18.04
Driver Version: 460.27   
CUDA Version: 11.1
torch:  1.10.2+cu111           
torchvision:  0.11.3+cu111
torchaudio:  0.10.2+cu111  
kornia:   0.6.5 

小坑

在docker里装了好几次，kornia的版本和torch及torchvision的版本总是不能匹配，废了不少时间

Solution：

conda create -n new_name --clone base_name  #搞个新环境试错、免得装坏了
# 先卸
pip uninstall torch
pip uninstall torchvision
# 先装kornia（默认最新）
pip install kornia
pip install kornia[x]
# 再装torch（根据显卡算力确定可行的cuda版本，cuda版本对应的算力要不高于GPU的算力，且跨版本号就不行，如3080 GPU算力sm_86, cuda版本要11以上；若装好了cuda就nvcc -V  看一下版本，再根据cuda版本装对应torch版本）
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111