ImportError: No module named ‘torch‘ 解決方法

鄒小妹發表於2020-10-28

原文連結:Anaconda+Pycharm環境下的PyTorch配置方法

https://blog.csdn.net/aa3615058/article/details/89339790?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-4.edu_weight&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-4.edu_weight

問題:

    torch包在pycharm裡面的匯入問題

描述: 在cmd端 import torch 沒問題,在pycharm裡匯入的時候不成功。執行程式碼的時候會報錯:   

import torch
ImportError: No module named 'torch'

本文Python版本為3.5。存在python版本和torch版本不一致的問題。

解決方法:

在cmd端建立新的conda虛擬環境

(官方文件
https://conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html)

step1: 輸入:

conda create --name pytorch-py36 --channel pytorch python=3.6 pytorch-cpu torchvision-cpu
  •  

step2:使用conda虛擬環境

activate pytorch-py36
python
  •  

step3: 在開啟的python命令列裡輸入

import torch
import torchvision
import numpy

不報錯則說明構建conda虛擬環境pytorch-py36成功。
注意:pytorch-py36預設在anaconda目錄下的envs資料夾中。

然後,在pycharm中使用conda虛擬環境:

將自己建立的conda虛擬環境pytorch-py36新增到pycharminterpreter可選列表中。

A:  在pycharm裡新建一個project.

  • 新建工程介面,選擇Existing enviroment,點選...按鈕,會彈出新增介面。
  • settings介面,點選齒輪按鈕,選擇Add,會彈出新增介面。
    在新增介面左側選擇Conda Environment,右邊選擇Existing environment,再點選...按鈕新增一個已經存在的conda虛擬環境,即剛剛構建的pytorch-py36

使用已經新增進來的虛擬環境pytorch-py36

B: 在interpreter列表中選擇剛剛新增的interpreter

如果需要對剛新增的interpreter改名,比如把python 3.6 (3)改成pytorch-py36,則點選齒輪按鈕,點選show all,裡面可以改名。

大功告成!

試著使用import torchimport torchvision這樣的語句,不會報錯則成功。

 

相關文章