Does Pytorch 1.7 support CUDA Toolkit 11.0 or CUDA Toolkit 11.0 Update1?

In this web page, there are two versions, CUDA Toolkit 11.0 and CUDA Toolkit 11.0 Update1 . CUDA Toolkit Archive | NVIDIA Developer
I am confused. If I want to install Pytorch version with CUDA, Which CUDA should I choose?

Double post from here with a potential answer.

  • you don’t need a local CUDA toolkit installation if you don’t plan on building PyTorch from source or any CUDA extension, just the NVIDIA driver.
  • the binaries use CUDA11.0.3, so 11.0 Update 1. If you plan on building custom CUDA extensions, you should install the same local toolkit