Hi @ppn, if you are installing PyTorch from pip, it won’t be built with CUDA support (it will be CPU only). We have pre-built PyTorch wheels for Python 3.6 (with GPU support) in this thread, but for Python 3.8 you will need to build PyTorch from source.
See this thread for further info about building PyTorch for Python 3.8: