From 338a8ae2d08d64f8c57c04eaebf6743f816031cd Mon Sep 17 00:00:00 2001
From: jonahpearl <68478436+jonahpearl@users.noreply.github.com>
Date: Mon, 6 Feb 2023 16:26:08 -0500
Subject: [PATCH] Pin torch to be below 1.13

Torch 1.13 has some weird extra NVIDIA dependencies, described [here] (https://github.com/pytorch/pytorch/issues/88049), which don't seem to recognize that cuda is loaded as a module. So as it currently stands, you need to install top-bottom and then uninstall those nvidia packages, and then maybe muck around a bit more so that everything points to the right place. I don't see any issue with pinning torch 1.12 for now.
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b65cd4c..203b244 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@
         'scikit-learn',
         'scipy',
         'segmentation_models_pytorch',
-        'torch',
+        'torch<1.13',
         'torchinfo',
         'torchplot',
         'torchvision',