Hey!
Did you follow the instructions in https://pytorch.org/get-started/locally/ to install inside your docker image?