update docs support torch

gyzhou2000 · gyzhou2000 · commit 7a64e766d31e · 2024-07-23T19:01:40.000+08:00
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -7,7 +7,9 @@ sphinx_copybutton==0.4.0
 numpy==1.24
 scipy
 tensorflow==2.11.0
-torch
+torch==2.2.2
+torchvision==0.17.2
+torchaudio==2.2.2
 tqdm
 hyperdash
 git+https://github.com/dddg617/TensorLayerX.git@nightly
