From 8fa27c861c5a7cbf8740cc911c9ff71316a9e151 Mon Sep 17 00:00:00 2001
From: Fatboy <87150253+gefei-qian-nih@users.noreply.github.com>
Date: Wed, 15 Nov 2023 09:22:26 -0500
Subject: [PATCH] Update SE3nv.yml

Pytorch-gpu is needed instead of pytorch. It will pull the dependencies on its own.
---
 env/SE3nv.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: