diff --git a/environment_cuda.yml b/environment_cuda.yml
index af1d7d3..1ecf55d 100644
--- a/environment_cuda.yml
+++ b/environment_cuda.yml
@@ -12,6 +12,7 @@ dependencies:
   - pip
   - pyg
   - python
+  - pytorch
   - pytorch-cluster
   - pytorch-scatter
   - pytorch-sparse