Merge branch 'bit-depth-power-example' of github.com:simplymathematic…

…s/deckard into bit-depth-power-example
simplymathematics · Dec 3, 2023 · b09f54c · b09f54c
2 parents 7c85be1 + b4a5655
commit b09f54c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/bit_depth/dvc.yaml b/examples/bit_depth/dvc.yaml
@@ -1,6 +1,6 @@
 stages:
   install_deckard:
-    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
+    cmd: python -m pip install -e ../../ && python -m pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html
     deps:
       - ../../setup.py
     outs: