onnxruntime/orttraining
Thiago Crepaldi 55c745eefd
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.
2022-09-08 15:30:44 -04:00
..
orttraining Add support for ORTModule Torch cpp CUDA extension build within docker (#12868) 2022-09-08 15:30:44 -04:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools Remove unused orttraining amd dockerfiles and scripts (#12707) 2022-09-02 18:43:21 -07:00