import torch ModuleNotFoundError: No module named 'torch'

ufoym / deepo

Setup and customize deep learning environment in seconds.

http://ufoym.com/deepo

MIT License

6.32k stars 750 forks source link

import torch ModuleNotFoundError: No module named 'torch' #150

Open GiveMeAnApple99 opened 1 year ago

GiveMeAnApple99 commented 1 year ago

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.