Hi,
You might have multiple pytorch installs coexisting and causing issues.
I would make sure to uninstall all pytorch installs (via conda / pip etc) in your environment and then reinstall a brand new one.