[doc] update torch install (wenet-e2e#1765)

* [doc] update torch install * fix
Mddct · Mar 20, 2023 · da09210 · da09210
1 parent f5b9b3e
commit da09210
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/install.md b/docs/install.md
@@ -25,7 +25,6 @@ git clone https://github.com/wenet-e2e/wenet.git
 conda create -n wenet python=3.8
 conda activate wenet
 pip install -r requirements.txt
-conda install pytorch=1.10.0 torchvision torchaudio=0.10.0 cudatoolkit=11.1 -c pytorch -c conda-forge
 ```
 
 ## Build for deployment

diff --git a/requirements.txt b/requirements.txt
@@ -14,3 +14,5 @@ flake8-pyi==20.5.0
 mccabe
 pycodestyle==2.6.0
 pyflakes==2.2.0
+torch==2.0.0
+torchaudio==2.0.0