diff --git a/env/SE3nv.yml b/env/SE3nv.yml index a51bcce6..f77127fa 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: