diff --git a/doc/environment.yml b/doc/environment.yml
index 97060c3004..635f24fe1e 100644
--- 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.*