Hi James!



 JamesOwers:

I’m looking for the minimal compute capability which each pytorch version supports.


Speaking from memory …
I went through a similar issue.  Based on what I was told, my
understanding is:
There is no table or clean record of which versions of pytorch
support which compute capabilities.  Even if a version of pytorch
uses a “cuda version” that supports a certain compute capability,
that pytorch might not support that compute capability.
The installation packages (wheels, etc.) don’t have the supported
compute capabilities encoded in there file names.
Pytorch has a supported-compute-capability check explicit in its
code.  (I’m not sure where.)  If you can figure out what version
of the source a given installation package was built from you
can check the code.
Short of that, I think you have to run pytorch and see whether
it likes your gpu.  (I’m not aware of a way to query pytorch for
its supported compute capability without running it against
the gpu of interest, but there might be one.)
Best.
K. Frank