Hi,
Is your PyTorch package built with CUDA support?
You can find our packages with CUDA support in the below link:
Thanks.