diff --git a/install/requirements_cuda.txt b/install/requirements_cuda.txt
index 71ba00c..ef3bb03 100644
--- a/install/requirements_cuda.txt
+++ b/install/requirements_cuda.txt
@@ -2,7 +2,7 @@ tqdm
 numpy
 matplotlib
 scipy
-networkx[all]
+networkx
 scikit-learn
 einops
 torch