From 75d8319286f20bebc22b7d1ebbb3949991e5a9ee Mon Sep 17 00:00:00 2001 From: liqunfu Date: Tue, 13 Apr 2021 20:36:24 -0700 Subject: [PATCH] Liqun/ort package name2 (#7337) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4ac1fbe6da..dec108f6e9 100644 --- a/setup.py +++ b/setup.py @@ -251,7 +251,7 @@ if enable_training: # this is needed immediately by pytorch/ort so that the user is able to # install an onnxruntime training package with matching torch cuda version. package_name = 'onnxruntime-training' - if cuda_version and not nightly_build: + if cuda_version: # removing '.' to make Cuda version number in the same form as Pytorch. cuda_version = cuda_version.replace('.', '') local_version = '+cu' + cuda_version