I am trying to build a docker which includes PyTorch starting from the L4T docker image. The docker build compiles with no problems, but when I try to import PyTorch in python3 I get this error:
Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/usr/local/lib/python3.6/dist-packages/torch/__init__.py", line 81, in <module> from torch._C import * ImportError: libnvToolsExt.so.1: cannot open shared object file: No such file or directory
I have installed PyTorch from the wheel provided here https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano-version-1-3-0-now-available/
Are there some dependencies I would need to include in my Dockerfile in order to have pytorch work?