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

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

  • 引言
  • 解决方案
  • pytorch 安装
  • 引言

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

    解决方案

    首先错误如下:

    解决:
    Ctrl+Z退出python,输入以下命令确认系统有pytorch

    conda env list
    

    如下图所示可以看到pytorch所在的环境名称为Pytorch-gpu

    接着输入以下命令进入环境

    conda activate Pytorch-gpu
    

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

    pytorch 安装

    1.cpu下的pytorch 安装

    pip install torch
    

    2.gpu下的pytorch 安装
    首先进入pytorch官网
    往下滑输入自己的对应各个环境的版本,底下那一行就是安装命令,打开命令行输入该命令即可

    注:也可以通过以下命令在单独创建一个环境,进入环境安装torch,实现上文中多个环境的效果
    创建环境,其中为pytorch1环境名可自定义

    conda create -n pytorch1 python=3.9
    

    进入环境

    conda activate pytorch1
    

    来源:Lin-CT

    物联沃分享整理
    物联沃-IOTWORD物联网 » 【解决方案】系统已经安装pytorch却调用不了,报错ModuleNotFoundError: No module named ‘torch‘

    发表评论