I am using jetson NX xavier kit having cuda 10.2.89, open Cv 4.1.1 and tensorRT 7.1.3 . Trying to install pytorch. Tried installing with this line
conda install pytorch torchvision cpuonly -c pytorch
but when i write this line
import torch
It throws an error by saying that module not installed.
How I can verify if pytorch has been installed correctly.
Hi @arsalanmajid116, please install one of the pip wheels from this topic:
These wheels are built with CUDA/cuDNN enabled.
Or you can use the l4t-pytorch or l4t-ml containers from NGC if that’s easier, both of which come with PyTorch and torchvision pre-installed.
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.