安装 kornia 版本问题

波斯猫_

已于 2022-06-22 12:25:34 修改

阅读量1.2w

点赞数 7

分类专栏：命令记录文章标签： pytorch 深度学习人工智能

于 2022-05-27 14:50:43 首次发布

本文链接：https://blog.csdn.net/mr1217704159/article/details/125003528

版权

命令记录专栏收录该内容

14 篇文章 0 订阅

订阅专栏

官方连接
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