安装lightly导致重装pytorch的问题

项目需要用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版的:

解决pytorch capability sm_86 is not compatible with the current PyTorch installation 问题_a563562675的博客-CSDN博客

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

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值