虚拟环境中pytorch及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

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

查看python版本


(2)pytorch安装

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

cuda安装
因为博主本人是要在服务器虚拟环境中安装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版本

查看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__)

结果如下:
请添加图片描述

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

  • 6
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值