Thank you @dusty_nv ! I’ve seen one of your older posts on how to install torchvision and I was planning to follow the instructions that you have provided!
I finally managed to install everything and have everything working, but I might give a try to the container as well! The only “issue” that I run into was that I can’t import OpenCV before importing Tensorflow.