Workspace Size Error by Multiple Conv+Relu Merging on DRIVE AGX

AI & Data Science Deep Learning (Training & Inference) TensorRT
7

Hi,

How did you manage to install pytorch on Pegasus AGX?
Regardless of the pytorch wheel file I install from this link, I keep getting this error when trying to import torch after installing the wheel file.

ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory