Virtual env picking up global pytorch version in Jetson

[vinayakah]

Hi, I have a NVIDIA Orin that has a global pytorch version = 1.12.0a0+02fb0b0f.nv22.06 installed. Jetpack version of Orin = 5.0.1.
I wanted to install PyTorch v1.13.0 in a venv. So, I created a venv and installed pytorch by using v1.13.0 from this link; https://developer.download.nvidia.cn/compute/redist/jp/v50/pytorch/

But, after installing and trying torch.__version__, somehow the venv is picking up the global v1.12.0 of pytorch instead of picking up v1.13.0.

The pip3 and python3 point to the venv site-packages and running pip3 list also lists the local venv modules. So, not sure why it’s picking up the global version ?

I have tried deactivating and reinstalling venv multiple times and with different versions of python as well !

Screenshots :

pytorch picks up global v1.12.0 not sure why ?

[AakankshaS]

Hi @vinayakah ,
This forum talks about issues/concerns related to cudnn.
Jetson forum might be helpful here.

Thanks

[AastaLLL]

Hi,

Do you also need help on this issue?
Or you prefer to upgrade to JetPack 5.1.1 and install PyTorch 2.0?

Thanks.