diff --git a/RF2na-linux.yml b/RF2na-linux.yml
index 6f737b3..425ffb0 100644
--- a/RF2na-linux.yml
+++ b/RF2na-linux.yml
@@ -7,7 +7,6 @@ channels:
 dependencies:
   - python=3.10
   - pip
-  - pytorch
   - requests
   - pytorch-cuda=11.7
   - dglteam/label/cu117::dgl
@@ -21,4 +20,4 @@ dependencies:
   - bioconda::csblast
   - pip:
     - psutil
-    - tqdm
\ No newline at end of file
+    - tqdm
diff --git a/SE3Transformer/requirements.txt b/SE3Transformer/requirements.txt
index cb6ce77..479d8a7 100644
--- a/SE3Transformer/requirements.txt
+++ b/SE3Transformer/requirements.txt
@@ -2,3 +2,5 @@ e3nn==0.3.3
 wandb==0.12.0
 pynvml==11.0.0
 git+https://github.com/NVIDIA/dllogger#egg=dllogger
+--find-links https://download.pytorch.org/whl/torch_stable.html
+torch==2.0.1+cu117