Is this the same issue as Cannot setup the runtime env - #9 by architkulkarni? Could you check the latest response there? It shows that installing pytorch requires an extra pip option.
Another idea, which is more of a guess: it looks like the pip install is failing in the middle. Usually the logs say Killed when this happens, I’m not sure why it didn’t show up in the logs though. Maybe try increasing the memory on the node? python - pip install - killed - Stack Overflow