安装 pytorch lighting
1 搜寻配对版本
- 进入lighting官网,查看配对版本
比如我就选择Python3.11、torch2.4、lightning2.4.0
2 搜寻pytorch安装命令
- 进入pytorch官网,查看以前版本的下载命令
- 注意要选择是 gpu版本的pytorch
- 查看自己显卡驱动命令:
nvidia-smi
- 查看pip命令,带有网站
3 自行下载,本地安装
- 直接输入命令安装会很慢,建议进入pip命令后面的网站,寻找到自己需要的包的版本,本地下载后,本地安装。比如我的网站就是:https://download.pytorch.org/whl/cu124
- 本地下载后,pip install 本地下载文件名 安装就可以了。
4 安装 lighting
安装指定版本的 lighting
pip install pytorch_lightning==2.4.0