Update build.yml

marble234 · May 7, 2021 · bf1ea23 · bf1ea23
1 parent dab13ab
commit bf1ea23
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/static/build.yml b/static/build.yml
@@ -5,9 +5,9 @@ dependencies:
     - ..  # d2l
     - git+https://github.com/d2l-ai/d2l-book
     - mxnet-cu101==1.7.0
-    - torch==1.8.0+cu101
+    - torch==1.8.1+cu101
     - -f https://download.pytorch.org/whl/torch_stable.html
-    - torchvision==0.9.0+cu101
+    - torchvision==0.9.1+cu101
     - -f https://download.pytorch.org/whl/torch_stable.html
     - tensorflow==2.3.1
     - tensorflow-probability==0.11.1