Не могу импортировать факел

Я установил pytorch, используя следующую команду

pip install --user torchvision

Но когда я пытаюсь импортировать его, я получаю ошибку

  Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  ImportError: No module named torch

Также он говорит, что все требования выполнены ]

    Requirement already satisfied: torchvision in ./.local/lib/python3.6/site-packages (0.2.1)
Requirement already satisfied: six in /usr/lib/python3/dist-packages (from torchvision) (1.10.0)
Requirement already satisfied: torch in ./.local/lib/python3.6/site-packages (from torchvision) (0.4.0)
Requirement already satisfied: pillow>=4.1.1 in ./.local/lib/python3.6/site-packages (from torchvision) (5.2.0)
Requirement already satisfied: numpy in ./.local/lib/python3.6/site-packages (from torchvision) (1.14.5)

Что мне делать?