Pytorch installation Issue

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.