From 83bd1f0db63cc515b038a895e197e03d52a1ed38 Mon Sep 17 00:00:00 2001
From: YellowRoseCx <80486540+YellowRoseCx@users.noreply.github.com>
Date: Mon, 20 Feb 2023 18:40:21 -0600
Subject: [PATCH] updates PyTorch to allow AMD gpu splitting

Feature adding update for AMD GPUs:
Changes the installed PyTorch version from 1.12.1+rocm5.1.1 to the newer stable version 1.13.1+rocm5.2 to allow AMD GPUs to utilize VRAM splitting across multiple GPUs
---
 environments/rocm.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/environments/rocm.yml b/environments/rocm.yml
index 9679d036..ad6ea39a 100644
--- a/environments/rocm.yml
+++ b/environments/rocm.yml
@@ -22,8 +22,8 @@ dependencies:
   - Pillow
   - psutil
   - pip:
-    - --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
-    - torch==1.12.1+rocm5.1.1
+    - --extra-index-url https://download.pytorch.org/whl/rocm5.2
+    - torch==1.13.1+rocm5.2
     - flask-cloudflared==0.0.10
     - flask-ngrok
     - flask-cors