Hello, I try to get a PyTorch v1.9.0 running in jupyterlab v3.1.13 after installation using the steps provided in the forum’s issue 72048 on a fresh JetPack4.6 SD. Tensorflow is 1.15.5+nv21.9 and dependencies are following the last pytorch nvidia container description.
Unfortunately, the sample scripts from PyTorch or others freeze or hang once the preprocessed image is transferred to the model(preprocessed image). Tensors images can be transferred to GPU, but scripted_predictor(batch) freezes the system. Error: /home/…/module.py:1051:UserWarning: Named tensors and all their associated APIs are an experimental feature etc. . (triggered internally at /media/nvidia/NVME/pytorch-v1.0.0/c10/core/TensorImpl.h:1156.) return forward_call(*input,**kwargs).
Would be glad to get some hints to improve the situation.