虚拟环境中pytorch及pytorch-lightning老版本安装_pytorch_lightning版本

虚拟环境中pytorch及pytorch-lightning安装：

博主本人最近在复现一篇论文，根据github中要求的依赖配置环境，要求如下图。从而引出本文的三个任务：

（1）python版本指定

（2）pytorch-lightning指定版本安装

（3）安装pytorch-lightning 1.7.2对应的pytorch

---

（1）python版本指定

创建虚拟环境时指定python版本即可

在创建虚拟虚拟环境时，可以直接指定python版本。比如，创建名为emp_bigfive，python版本为3.9.1的虚拟环境

conda create -n emp_bigfive python=3.9.1
1

如此，创建好的虚拟环境中，python版本就为指定版本。查看python版本指令如下：

---

（2）pytorch安装

要复现的论文中未指定cuda版本，只指定了pytorch-lightning版本。按照pytorch-lightning与pytorch版本对应关系选择了pytorch 1.11.0

因为博主本人是要在服务器虚拟环境中安装pytorch，故使用如下代码：

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
1

---

（3）pytorch-lightning安装：

这里最困难的就是要安装指定版本的pytorch-lightning

如果不指定版本，代码如下。安装很快

conda install pytorch-lightning -c conda-forge
1

指定了1.7.2版本的pytorch-light后，语句如下:

conda install pytorch-lightning==1.7.2 torchmetrics==0.11.4 -c conda-forge
1

这里下载很慢，耐心等待
注意上面语句中必须要带torchmetrics==0.11.4，否则安装成功后，使用import pytorch_lightning时会报错！！！（别问我怎么知道的，问就是我一个朋友遇到过这个问题┭┮﹏┭┮）

检查是否成功安装

①检查python版本

②检查pytorch版本

import torch
print(torch.__version__)     #检查torch版本
torch.cuda.is_available()    #输出为true说明安装的是gpu版本
1
2
3

结果如下：

③检查pytorch-lightning版本

import pytorch_lightning     #注意不是pytorch-lightning!!!不是-而是_
print(pytorch_lightning.__version__)
1
2

结果如下：

祝大家都能顺利配好环境，多成功少失败~~~