Pytorch 0.4 get cuda version

I want to compile a cuda extension, using setup.py and CUDAExtension as described here.
this thread is similar to my question but not exactly.

How can I get the cuda version that my “conda install pytorch” is currently using? I want to make sure the binaries would be compatible.

4 Likes

I figured out:

torch.version.cuda

Perhaps this should be mentioned in the document

35 Likes