Hi,
Please install the PyTorch package that has been built with CUDA support. You can find the package in the topic below:
Thanks.