Update supported TF versions, update setup_packages.py (#1693)

- uses only one 1.12.0 version of TF for CUDA 9 - removes 1.13.1 for CUDA 10 and adds 2.1.0 - adjustes setup_packages.py to match the newest pip version Signed-off-by: Janusz Lisiecki <[email protected]>
msr-fiddle · Jan 31, 2020 · 410cb04 · 410cb04
1 parent 42d8117
commit 410cb04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -29,8 +29,8 @@
                         "90" : ["1.5.0"],
                         "100" : ["1.5.0"]},
             "tensorflow-gpu" : {
-                "90": ["1.12.0", "1.11", "1.7"],
-                "100": ["1.13.1", "1.14.0", "1.15.0", "2.0.0"]},
+                "90": ["1.12.0",],
+                "100": ["1.14.0", "1.15.2", "2.0.1", "2.1.0"]},
             "torch" : {"90": ["http://download.pytorch.org/whl/{cuda_v}/torch-1.1.0-{0}.whl"],
                        "100": ["http://download.pytorch.org/whl/{cuda_v}/torch-1.2.0-{0}.whl"]},
             "torchvision" : {"90": ["https://download.pytorch.org/whl/{cuda_v}/torchvision-0.3.0-{0}.whl"],