i have to ask admins to upgrade to torch 1.10.0 and if possible to nightly.
they rebuild pytorch to optimize it on the hardware of the cluster.
i’ll let you know.
i already asked to upgrade to 1.10.0 (+torchvision) couple days ago. it takes time…
i dont want to install from pypi as it may not reproduce the same error.
and it is recommended, by admins, to install prebuilt (on cluster) packages.
thanks