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.