-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Add torch.__config__.show(), reporting detailed version of all libraries. #18579
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
…all libraries. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]>
|
this fixes #111 You might want to rename the methods slightly to match numpy, if you think it makes sense |
|
Makes sense to me! |
dzhulgakov
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Would be nice to unify it with Caffe2's build options here: caffe2/core/macros.h.in caffe2/core/common.cc
Specifically, if you follow new format, I'd add the following compilation flags:
- BUILD_TYPE
- CXX_FLAGS
- BLAS (as we will likely replace the ugly THBlas soon)
- USE_NNPACK
- USE_FBGEMM (as we're adding quantization support now) - and include its version
torch/csrc/Module.cpp
Outdated
| #endif | ||
| } | ||
|
|
||
| static PyObject *THPModule_detailedVersion(PyObject *module) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nit: shall we start using pybind universally?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
For things like this one sure, but I don't trust pybind enough to feel ok with using it for our performance-critical eager functions.
Thanks, was not aware of this. |
fbgemm does not appear to have any useful versions. Determined by looking at |
… all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
…ailed version of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
…iled version of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
…sion of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
|
For cuda/cudnn, etc, could build-time version be different from run-time version in its minor version number or patch number? |
|
@ppwwyyxx Makes sense to me! |
… all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
|
Now also with nvcc arch information |
… all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
…detailed version of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
…etailed version of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
dzhulgakov
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
So, do you want to also include the following:
BUILD_TYPE
CXX_FLAGS
BLAS (as we will likely replace the ugly THBlas soon)
USE_NNPACK
USE_FBGEMM (as we're adding quantization support now) - and include its version
Even if FBGEMM doesn't have exported version, just the fact whether we build with it or not, same for NNPACK.
Even for MKL and others - I'd suggest to explicitly adding a line "MKL: none" to the output
|
@dzhulgakov I'm assuming part of your comment is an accidental copy paste from the earlier comment. Right now, I print all of the existing Caffe2 settings as part of the |
… detailed version of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
dzhulgakov
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Apologies, I was looking at the old code version (GH UI is confusing)
…ch.__config__.show(), reporting detailed version of all libraries." Add torch.__config__.show(), reporting detailed version of all libraries. Fixes #111. Example output: ``` PyTorch built with: - GCC 4.8 - Intel(R) Math Kernel Library Version 2019.0.3 Product Build 20190125 for Intel(R) 64 architecture applications - Intel(R) MKL-DNN v0.18.1 (Git Hash 7de7e5d02bf687f971e7668963649728356e0c20) - OpenMP 201107 (a.k.a. OpenMP 3.1) - CUDA Runtime 9.2.0 - CuDNN 7.1.2 ``` Signed-off-by: Edward Z. Yang <[email protected]> gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
|
would this be useful for |
…ies. (#18579) Summary: Pull Request resolved: pytorch/pytorch#18579 ghimport-source-id: 65124c95e49423de4ad1008c65e75057fea09b94 Differential Revision: D14778507 Pulled By: ezyang fbshipit-source-id: 1e4bb79f4800a116ce8fb7af2fefbd34da8d102c
There is no cudnn 7.10.2. Cudnn version numbers work a bit different from cuda. 7102 means 7.1.2. |
|
@ppwwyyxx Oops! Thanks for the catch. |
…ies. (pytorch#18579) Summary: Pull Request resolved: pytorch#18579 ghimport-source-id: 65124c9 Differential Revision: D14778507 Pulled By: ezyang fbshipit-source-id: 1e4bb79f4800a116ce8fb7af2fefbd34da8d102c
Stack from ghstack:
Fixes #111.
Example output:
Signed-off-by: Edward Z. Yang [email protected]
gh-metadata: pytorch pytorch 18579 gh/ezyang/48/head
Differential Revision: D14778507