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:
diff --git a/setup.py b/setup.py
index 38e2fcc0..bcb0392c 100644
--- a/setup.py
+++ b/setup.py
@@ -1,10 +1,10 @@
 from setuptools import setup, find_packages
 
 setup(name='rfdiffusion',
-      version='1.1.0',
+      version='1.1.1',
       description='RFdiffusion is an open source method for protein structure generation.',
       author='Rosetta Commons',
       url='https://github.com/RosettaCommons/RFdiffusion',
       scripts=["scripts/run_inference.py"],
       packages=find_packages(),
-      install_requires=['torch==2.1.2', 'se3-transformer'])
+      install_requires=['torch', 'se3-transformer'])