Pytorchのインポートができない

Question

前提・実現したいこと

Anacondaで構築した仮想環境で"Pytorch"を使いたい。

発生している問題・エラーメッセージ

仮想環境上からJupyter Notebookを開き、

import torch

としたところ、

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-9-63e830a3eae1> in <module>()
----> 1 import torch
      2 #torch.cuda.is_available()
      3 #torch.cuda.current_device()

ModuleNotFoundError: No module named 'torch'

となってしまいました。

確認事項

仮想環境上で

conda list

したところ、

pytorch                   1.7.1           py3.8_cuda101_cudnn7_0    pytorch

の記載を確認しております。

一方、"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include\cudnn.h"を確認したところ、

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5

となっておりました。

Comments

[episteme]

コマンドラインからpythonを起動し、import pytorch しても失敗しますか?

[jbpb0]

Jupyter Notebook で import torch がうまくいかない状態で (Jupyter Notebook 上で)
!conda list torch
を実行したら、pytorch とか出てきますか?

[Y_Miz]

epistemeさん、
Anaconda Promptで仮想環境に入り、
"import torch
print(torch.cuda.is_available())"
と記述した.pyファイルを実行したら、エラーとならず、うまくいきました（"True"と出力）。

もちろん、コマンドプロンプト上で、上記を実行してもうまくいきました。

ただし、Jupyter Notebook上ではうまくいっていません。

[Y_Miz]

jbpb0さん、
以下のように出力されました。
ーーーーーーーーーーーーーーーー
# packages in environment at C:\Users\AppData\Local\conda\conda\envs\pytorch:
#
# Name Version Build Channel
pytorch 1.7.1 py3.8_cuda101_cudnn7_0 pytorch
torchaudio 0.7.2 py38 pytorch
torchvision 0.8.2 py38_cu101 pytorch
ーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーー

なお、仮想環境上（Anaconda Prompt）で"pip show torch"を実行したら、
下記のように出力されました。
ーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーーー
Name: torch
Version: 1.7.1
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: [e-mail]
License: BSD-3
Location: c:\users\appdata\local\conda\conda\envs\pytorch\lib\site-packages
Requires: typing-extensions, numpy
Required-by: torchvision, torchaudio

[jbpb0]

仮想環境のJupyter Notebookで
!pip show torch
しても、Anacondaプロンプトの場合と同じのが表示されると思いますが、念の為に確認してみてください
(特に「Location:」が同じか)

あと、仮想環境のJupyter Notebookで
import sys
import pprint
pprint.pprint(sys.path)
して表示されるパスの中に、
c:\users\appdata\local\conda\conda\envs\pytorch\lib\site-packages
が入ってますでしょうか? (フォルダ区切りの「\」は二つ並んでいると思います)

【追記】解決した、というのを読む前に書いてしまいました

Answer 1

仮想環境下で、"conda install jupyter"を実行したら、jupyter notebook上でも通りました。
お騒がせいたしました。