From a7099e2edac0988352af82a4a8aefeda3aafd82f Mon Sep 17 00:00:00 2001 From: chrischoy Date: Mon, 15 Nov 2021 09:27:28 -0800 Subject: [PATCH] torch version string man fix --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 04c051c5..a2826848 100644 --- a/setup.py +++ b/setup.py @@ -147,7 +147,7 @@ def _argparse(pattern, argv, is_flag=True, is_list=False): CC_FLAGS = [] NVCC_FLAGS = [] -TORCH_VERSION = [int(v) for v in torch.__version__.split('.')] +TORCH_VERSION = [int(v) for v in torch.__version__.replace('+', '.').split('.')[:3]] if CPU_ONLY: print("--------------------------------")