What are the commands needed to install pytorch 1.7 with torchvision 0.8.1 for cuDNN 10.2 in Jetson Xavier NX?
I read in this forum post PyTorch for Jetson - version 1.7.0 now available that 1.7 is available but from past experiences without matching the correct pytorch version with torchvision and cuDNN, running a computer vision model would generate the the Segmentation Fault (core dumped) issue.
At the moment I am running Pytorch 1.6 with torchvision 0.7 but I am quite sure there is a compatibility issue as after executing the CNN model in pytorch, the program does not close correctly and shows a Segmentation Fault (core dumped) error. I read in a different post that, this issue could be caused by Pytorch not coupled with the correct torchvision library and/or cuDNN version.
As I am relatively new to using the ubuntu environment and using jetson nx, I could not make sure if the commands listed above forum would install pytorch 1.7 with the correct version of torchvision and will work in cuDNN 10.2
** Jetson Xavier NX running Jetpack 4.4.1
** OS - Ubuntu 18.04 LTS