From 1b9a67714d49b68d470d7c1aab8f6e65d07cc6fc Mon Sep 17 00:00:00 2001
From: simplymathematics <15224968+simplymathematics@users.noreply.github.com>
Date: Wed, 29 Nov 2023 01:57:59 +0100
Subject: [PATCH] Reproduce (#142)

-- GPU bug for torch models
---
 examples/power/dvc.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/power/dvc.yaml b/examples/power/dvc.yaml
index b20ecef7..f4023a65 100644
--- a/examples/power/dvc.yaml
+++ b/examples/power/dvc.yaml
@@ -1,6 +1,6 @@
 stages:
   install_deckard:
-    cmd: python -m pip install -e ../../ && python -m pip install torch torchvision
+    cmd: python -m pip install -e ../../ && python -m pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
     deps:
       - ../../setup.py
     outs: