From 41b182e134d82047e6251975e8e0a072683ef462 Mon Sep 17 00:00:00 2001
From: Pete MacKinnon <mackinnon.pd@gmail.com>
Date: Sat, 28 Jan 2023 01:30:38 -0400
Subject: [PATCH] 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: https://github.com/nv-morpheus/Morpheus/pull/645
---
 docker/conda/environments/requirements.txt                  | 6 +++---
 .../phishing-models/requirements.txt                        | 2 +-
 .../root-cause-models/requirements.txt                      | 2 +-
 .../dfp-models/requirements.txt                             | 2 +-
 .../phishing-models/requirements.txt                        | 2 +-
 .../root-cause-models/requirements.txt                      | 2 +-
 setup.py                                                    | 4 ++--
 7 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/docker/conda/environments/requirements.txt b/docker/conda/environments/requirements.txt
index 6970ecdc4f..17540f4c64 100644
--- 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
index a6745592d8..fc6905b410 100644
--- 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
index a6745592d8..fc6905b410 100644
--- 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
index 802bd43663..451e0b4d2e 100644
--- 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
index ad4660fb54..5d41fef50c 100644
--- 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
index ad4660fb54..5d41fef50c 100644
--- 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
index e614e66c0c..2e83cf538e 100644
--- 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",