Lei_Yang:

Do I need to install cuda and cudnn that matches the version of cudatoolkit shipped with pytorch conda install?


You don’t need a local CUDA installation, if you don’t plan to build PyTorch from source or any custom CUDA extension (or of course any other CUDA application).
That being said, could you update to the latest stable release and check, if you are still facing this issue?