onnxruntime
55c745ee - Add support for ORTModule Torch cpp CUDA extension build within docker (#12868)

Commit
3 years ago
Add support for ORTModule Torch cpp CUDA extension build within docker (#12868) Currently, CUDA hardware is not available to be leveraged by build during `docker build`. because of that, CUDA capable hardware would not have CUDA support This PR adds an env varf ONNXRUNTIME_FORCE_CUDA in which it allows CUDA extensions to be compiled even when CUDA support is not detected.
Author
Thiago Crepaldi
Parents
Loading