官网地址:
安装PyTorch-Lightning
1、不能直接使用pip install pytorch-lightning ,否则如下图会直接卸载掉你的torch而安装cpu版本的torch。
Installing collected packages: torch, lightning_fabric
Attempting uninstall: torch
Found existing installation: torch 1.9.1+cu111
Uninstalling torch-1.9.1+cu111:
Successfully uninstalled torch-1.9.1+cu111
2、在安装pytorch-lightning时一定注意自己的torch是pip安装还是conda安装,两者要保持一致,否则也会导致你的torch版本被替换。
pip install pytorch-lightning==版本名
或:
conda install lightning==版本名 -c conda-forge
验证PyTorch和PyTorch-Lightning版本
在Python中导入PyTorch,然后打印出其版本号来查看,示例代码如下:
import torch
print(torch.__version__)
使用以下代码来查看已安装的 PyTorch-Lightning 版本:
import pytorch_lightning as pl
print(pl.__version__)
验证GPU是否能用
import torch
if torch.cuda.is_available():
print("CUDA is available")
else:
print("CUDA is not available")
import pytorch_lightning as pl
print(pl._C.has_cuda) # will print True if GPU is supported, False otherwise