From 7a64e766d31e348ba7dc74251569995c5cbc9eec Mon Sep 17 00:00:00 2001
From: gyzhou2000 <zhougy@bupt.edu.cn>
Date: Tue, 23 Jul 2024 19:01:40 +0800
Subject: [PATCH] update docs support torch

---
 docs/requirements.txt | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index 2d1ff16a..43083803 100644
--- 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