Find version of cuda library?

When pytorch gets installed, it will always also install its own cuda libraries for a specific cuda version, and for one pytorch version, it is possible to install that pytorch version with various versions of the cuda library included.

I have pytorch version 1.7.1, how can I figure out from within python what the version of the cuda library that was installed with it is?

Does torch.version.cuda work?

1 Like

Yes that worked, thank you!
I tried all variations of torch.cuda.__version__ instead :confused: