diff --git a/conda_env_tomotwin.yml b/conda_env_tomotwin.yml
index cc7a976..abfb72d 100644
--- a/conda_env_tomotwin.yml
+++ b/conda_env_tomotwin.yml
@@ -1,11 +1,13 @@
 name: tomotwin
 channels:
   - nvidia
-  - pytorch-nightly
+  - pytorch
   - rapidsai
   - conda-forge
   - defaults
 dependencies:
+  - pytorch[version='>=2.1']
+  - torchvision
   - pandas[version='<2']
   - scipy
   - numpy
@@ -19,8 +21,5 @@ dependencies:
   - mysql-connector-python
   - pip
   - pip:
-      - --extra-index-url https://download.pytorch.org/whl/nightly/cu118
-      - torch==2.1.0.dev20230822+cu118
-      - torchvision==0.16.0.dev20230822
       - tomotwin-cryoet