环境win11,anaconda虚拟环境下
查看自己的nvidia-smi信息,查看pytorch-lightning信息。我们的需求是CUDA版本,torch版本,lightning版本相匹配。
安装pytorch
我的nvidia-smi信息如下:

为了和lightning版本匹配,使用命令安装torch2.4.0
conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=11.8 -c pytorch -c nv