当前位置:   article > 正文

【解决方案】系统已经安装pytorch却调用不了,报错ModuleNotFoundError: No module named ‘torch‘_linux装好pytorch为什么第二次进去没有显示

linux装好pytorch为什么第二次进去没有显示

【解决方案】系统已经安装pytorch却调用不了,报错ModuleNotFoundError: No module named 'torch'

引言

云服务器上配置时显示已经有pytorch环境但是运行却报错说没有,这是由于没有进入pytorch所配置的环境造成的,进入对应环境即可运行pytorch

解决方案

首先错误如下:
在这里插入图片描述
解决:
Ctrl+Z退出python,输入以下命令确认系统有pytorch

conda env list
  • 1

如下图所示可以看到pytorch所在的环境名称为Pytorch-gpu
在这里插入图片描述
接着输入以下命令进入环境

conda activate Pytorch-gpu
  • 1

左侧出现(Pytorch-gpu)表示目前在Pytorch-gpu环境下,这是导入torch就没问题了
在这里插入图片描述

pytorch 安装

1.cpu下的pytorch 安装

pip install torch
  • 1

2.gpu下的pytorch 安装
首先进入pytorch官网
往下滑输入自己的对应各个环境的版本,底下那一行就是安装命令,打开命令行输入该命令即可
在这里插入图片描述
注:也可以通过以下命令在单独创建一个环境,进入环境安装torch,实现上文中多个环境的效果
创建环境,其中为pytorch1环境名可自定义

conda create -n pytorch1 python=3.9
  • 1

进入环境

conda activate pytorch1
  • 1
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小舞很执着/article/detail/794520
推荐阅读
相关标签
  

闽ICP备14008679号