pytorch-lightning环境配置记录(win11+anaconda)

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

安装pytorch

我的nvidia-smi信息如下:
pytorch-lightning环境配置记录(win11+anaconda)_第1张图片

为了和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

你可能感兴趣的:(pytorch,python)