How to use pytorch on a single node?