Update torch to 1.13.1+cu116 (#645)

Fixes #644 Authors: - Pete MacKinnon (https://github.com/pdmack) Approvers: - Eli Fajardo (https://github.com/efajardo-nv) - Michael Demoret (https://github.com/mdemoret-nv) URL: #645
nv-morpheus · Jan 28, 2023 · 41b182e · 41b182e
1 parent 2404b8d
commit 41b182e
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 10 deletions.
diff --git a/docker/conda/environments/requirements.txt b/docker/conda/environments/requirements.txt
@@ -1,6 +1,6 @@
 # Note, to include this when using setup.py or pip, set the variable:
-# PIP_FIND_LINKS=https://download.pytorch.org/whl/cu113/torch_stable.html
---find-links https://download.pytorch.org/whl/cu113/torch_stable.html
+# PIP_FIND_LINKS=https://download.pytorch.org/whl/cu116/torch_stable.html
+--find-links https://download.pytorch.org/whl/cu116/torch_stable.html
 
 ####### Pip-only runtime dependencies (keep sorted!) #######
 # Packages listed here should also be listed in setup.py
@@ -10,6 +10,6 @@ jupyter-core>=4.11.2,<5.0
 jupyterlab
 nvidia-pyindex
 pytest-benchmark>=4.0
-torch==1.12.0+cu113
+torch==1.13.1+cu116
 tritonclient[all]==2.17.*
 websockets
diff --git a/models/training-tuning-scripts/phishing-models/requirements.txt b/models/training-tuning-scripts/phishing-models/requirements.txt
@@ -3,6 +3,6 @@ cupy==10.6.0
 numpy==1.22.4
 pandas==1.3.5
 scikit_learn==1.1.3
-torch==1.12.0+cu113
+torch==1.13.1+cu116
 tqdm==4.64.1
 transformers==4.24.0
diff --git a/models/training-tuning-scripts/root-cause-models/requirements.txt b/models/training-tuning-scripts/root-cause-models/requirements.txt
@@ -3,6 +3,6 @@ cupy==10.6.0
 numpy==1.22.4
 pandas==1.3.5
 scikit_learn==1.1.3
-torch==1.12.0+cu113
+torch==1.13.1+cu116
 tqdm==4.64.1
 transformers==4.24.0
diff --git a/models/validation-inference-scripts/dfp-models/requirements.txt b/models/validation-inference-scripts/dfp-models/requirements.txt
@@ -6,4 +6,4 @@ matplotlib==3.4.2
 numpy==1.22.0
 pandas==1.0.1
 scikit_learn==1.0.2
-torch==1.12.0+cu113
+torch==1.13.1+cu116
diff --git a/models/validation-inference-scripts/phishing-models/requirements.txt b/models/validation-inference-scripts/phishing-models/requirements.txt
@@ -2,4 +2,4 @@ cudf==22.8.1
 numpy==1.22.4
 onnxruntime==1.13.1
 scipy==1.9.1
-torch==1.12.0+cu113
+torch==1.13.1+cu116
diff --git a/models/validation-inference-scripts/root-cause-models/requirements.txt b/models/validation-inference-scripts/root-cause-models/requirements.txt
@@ -2,4 +2,4 @@ cudf==22.8.1
 numpy==1.22.4
 onnxruntime==1.13.1
 scipy==1.9.1
-torch==1.12.0+cu113
+torch==1.13.1+cu116
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
 
 # Required to install torch via setup.py
 # Note: this is order dependent
-os.environ["PIP_FIND_LINKS"] = "https://download.pytorch.org/whl/cu113/torch_stable.html"
+os.environ["PIP_FIND_LINKS"] = "https://download.pytorch.org/whl/cu116/torch_stable.html"
 
 import versioneer  # noqa: E402
 from setuptools import find_packages  # noqa: E402
@@ -49,7 +49,7 @@
         # Only list the packages which cannot be installed via conda here. Should mach the requirements in
         # docker/conda/environments/requirements.txt
         "dfencoder @ git+https://github.com/nv-morpheus/dfencoder.git@branch-23.01#egg=dfencoder",
-        "torch==1.12.0+cu113",
+        "torch==1.13.1+cu116",
         "tritonclient[all]==2.17.*",  # Force to 2.17 since they require grpcio==1.41 for newer versions
     ],
     license="Apache",