Karthik_Ganesan:

The cuda version mistmatch might be causing this, since PyTorch seems to be using 10.1 but your system seems to have 11.4 installed.


A newer driver should be able to run older CUDA runtimes and the local CUDA toolkit won’t be used when the pip wheels or conda binaries are installed as they ship with their own runtime.
@sharyat_singh could you update to the latest PyTorch release (1.10.1 or the nightly binary) and check if you are still hitting the issue?