diff --git a/environment_cuda.yml b/environment_cuda.yml
index 1ecf55d..ca69ecb 100644
--- a/environment_cuda.yml
+++ b/environment_cuda.yml
@@ -29,6 +29,7 @@ dependencies:
   - gcsfs
   - pytest
   - pip:
+      - setuptools
       - datasets
       - einops
       - fsspec