For transfer learning : https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-collect.md
I have installed pytorch using ./install-pytorch.sh. for python3.6.
When I run in my home directory
I can see these both get imported without any errors.
While re-training model at :
$ cd jetson-inference/python/training/classification
$ python3 train.py --model-dir=(YOUR-MODEL) (PATH-TO-YOUR-DATASET)
I get the error : No module named ‘torchvision’
What could be the possible reason? Do I have to install torchvision separately?