由于安装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