How to use pytorch in this challenge?

I’ve tried to add torch to the requirements.txt but my estimator still fails with: “ModuleNotFoundError: No module named ‘torch’”.

I’ve also tried to add numpy only and get the same issue for numpy. So I guess I’m doing something terribly wrong. For the numpy only solution my requirements.txt looks like this:
numpy==2.4.4

also tried without pinning the version but got the same issue.