ufoym / deepo

Setup and customize deep learning environment in seconds.

Home Page:http://ufoym.com/deepo

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

import torch ModuleNotFoundError: No module named 'torch'

GiveMeAnApple99 opened this issue · comments

I used the command
docker run --gpus all --name MyName -p ******:22 -it -v /home/tom/DISK/DISK1/MyDirectory/:/home/ ufoym/deepo /bin/bash
to create a container based on the image ufoym/deepo.
But when I tried to import torch...

import torch
Traceback (most recent call last):
File "", line 1, in
ModuleNotFoundError: No module named 'torch'

So how can I solve this problem?
Thank u.