虚拟环境中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
如此,创建好的虚拟环境中,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
(3)pytorch-lightning安装:
这里最困难的就是要安装指定版本的pytorch-lightning
如果不指定版本,代码如下。安装很快
conda install pytorch-lightning -c conda-forge
指定了1.7.2版本的pytorch-light后,语句如下:
conda install pytorch-lightning==1.7.2 torchmetrics==0.11.4 -c conda-forge
这里下载很慢,耐心等待
注意上面语句中必须要带torchmetrics==0.11.4
,否则安装成功后,使用import pytorch_lightning
时会报错!!!(别问我怎么知道的,问就是我一个朋友遇到过这个问题┭┮﹏┭┮)
检查是否成功安装
①检查python版本
②检查pytorch版本
import torch
print(torch.__version__) #检查torch版本
torch.cuda.is_available() #输出为true说明安装的是gpu版本
结果如下:
③检查pytorch-lightning版本
import pytorch_lightning #注意不是pytorch-lightning!!!不是-而是_
print(pytorch_lightning.__version__)
结果如下: