解决pytorch-lightning1.6.0安装时自动下载最新的pytorch方法

Isaacingrid2025-08-07 13:35

由于安装lightning的时候会自动下载最新版本的pytorch，因此换了一种方法安装

Step 0. 无依赖下安装

py 复制代码

pip3 install pytorch-lightning==1.6.0 --no-deps

如果找不到1.6.0，则降低pip版本，如：

py 复制代码

pip install pip==24.0

Step 1. 安装lightning需要的依赖包，补充依赖

py 复制代码

pip3 install torchmetrics==0.7.0
pip install fsspec[http]==2021.05.0 
pip install tensorboard==2.14.0

Step 2.验证

py 复制代码

python -c "import torch; import pytorch_lightning as pl; \
print(f'PyTorch: {torch.__version__}\nPyTorch-Lightning: {pl.__version__}')"

Step 3. 检查

py 复制代码

pip3 list | grep -E "pytorch|torch|lightning|numpy|tensorboard"

py 复制代码

mmdet3d                 1.0.0rc4     /data/hx/project/BEVUDA-Pytorch/mmdetection3d-1.0.0rc4
numpy                   1.22.0
pytorch-lightning       1.6.0
tensorboard             2.14.0
tensorboard-data-server 0.7.2
tensorboard-plugin-wit  1.8.1
tensorboardX            2.6.2.2
torch                   1.9.0+cu111
torchaudio              0.9.0
torchmetrics            0.7.0
torchvision             0.10.0+cu111