diff --git a/env/SE3nv.yml b/env/SE3nv.yml
index a51bcce..f77127f 100644
--- a/env/SE3nv.yml
+++ b/env/SE3nv.yml
@@ -2,7 +2,7 @@ name: SE3nv
 channels:
   - defaults
   - conda-forge
-  - pytorch
+  - pytorch-gpu
   - dglteam
   - nvidia
 dependencies: