Is there a table which shows the supported cuda version for every pytorch version

Is there a table which shows the supported cuda version for every pytorch version?
Thanks.

You can build PyTorch from source with any CUDA version >=9.2 and the binaries ship with the mentioned CUDA versions from the install selection.

1 Like

If I only have cuda9.0, what’s the newest pytroch I can use?

I think 1.4 would be the last PyTorch version supporting CUDA9.0.
Note that you don’t need a local CUDA toolkit, if you install the conda binaries or pip wheels, as they will ship with the CUDA runtime. You would only have to make sure the NVIDIA driver is updated to the needed version corresponding to the CUDA runtime version.

1 Like

Thank you for your reply.

If I have to build some additional libraries, such as detectron2, do I need a local CUDA toolkit?

If these 3rd party packages are shipping with CUDA kernels, then you would need to install a local CUDA toolkit, yes.

1 Like