项目需要用lightly。
安装过程中发现依赖pytorch-lightning,本虚拟环境没有,所以会自动安装最高版本,然后会自动安装最高版本torch,因为cuda(11.0)关系,最新torch(已装1.7.1)并不是想要的,。
(安装模块,依赖模块中如果有满足最低要求的,就不会安装了,如果找不到满足最低要求的,会默认安装最新版)
(查看模块版本小技巧:版本高一点的pip有一个命令可以查看某一模块的所有版本:
pip index versions module_name)
pytorch-lightning官网没有要求的torch版本对应关系,自己手动试的(笨方法:pip安装pytorch-lightning,然后到下载torch的时候ctrl+c切断)
pytorch-lightning2.01 torch>=1.11.0
pytorch-lightning1.76 torch>=1.9
pytorch-lightning1.5.10 torch>=1.7.*
但安装1.5.10版本的时候还是会下载最新的,提示如下:
Collecting torch>=1.7.* (from pytorch-lightning==1.5.10)(这句提示是因为pytorch-lightning才重装的torch)
Downloading torch-2.0.0-cp38-cp38-manylinux1_x86_64.whl (619.9 MB)
(这里的torch包没有cu110这样的字眼,是cpu版的。而且我又建了个虚拟环境试了,装的最新torch,不能用cuda,要看nvcc版本对应的torch,而且要装cuda版的:
torch安装包地址:https://download.pytorch.org/whl/cu110/torch_stable.html
)
找到有人说是torchmetrics的原因,
然后又试torchmterics与torch对应关系
torchmterics1.11.1 对应torch>=1.8.1
torchmterics0.10.3 对应torch>=1.3.1
最后安了torchmterics0.10.3,
然后控制了两个变量,安了pytorch-lightning 1.5.9,因为感觉和1.5.10差不多,不想再试1.5.10了,也是 torch>=1.7.*,成功安装。
所以到底是pytorch-lightning还是torchmeterics的原因,大家可以自己试试,我不想努力了。
最后安装Lightly,1.3、1.4 版本(1.4是当前最新的)要求lightning>=1.0.4,经过前面试,符合,所以安装最新版Lightly即可。
时间:2023.04.24