use the latest torch torchvision torchaudio

simplymathematics · Dec 2, 2023 · b4a5655 · b4a5655
1 parent 076dce7
commit b4a5655
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: