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: