docs: install pytorch in RTD (#3333)

Fix No module named 'torch' --------- Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 24, 2024 · 03ca9ab · 03ca9ab
1 parent 649fdca
commit 03ca9ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doc/environment.yml b/doc/environment.yml
@@ -7,7 +7,7 @@ dependencies:
   - python=3.9
   - pip>=20.1
   - pip:
-    - ..[docs,cpu]
+    - ..[docs,cpu,torch]
     - "exhale @ https://github.com/svenevs/exhale/archive/2759a394268307b88f5440487ae0920ee4ebf81e.zip"
     # https://github.com/mcmtroffaes/sphinxcontrib-bibtex/issues/309
     - docutils!=0.18.*,!=0.19.*