Hi,
Suppose you install PyTorch from pip3?
Is that correct? The package doesn’t enable CUDA.
We don’t provide a Python 3.8 prebuilt PyTorch package for JetPack 4.
Please build it from the source to get the GPU support.
You can find the instructions below:
Thanks.