diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
index d2bdf61..3ef5303 100644
--- a/.devcontainer/Dockerfile
+++ b/.devcontainer/Dockerfile
@@ -30,6 +30,7 @@ ARG CUDATOOLKIT_VERSION=12.4
 RUN conda install pytorch=${PYTORCH_VERSION} pytorch-cuda=${CUDATOOLKIT_VERSION} -c pytorch -c nvidia
 
 # Handle environment.yml if it exists
+RUN echo env_change_20241021
 COPY environment.yml* noop.txt /tmp/conda-tmp/
 RUN if [ -f "/tmp/conda-tmp/environment.yml" ]; then \
     /opt/conda/bin/conda env update -n base -f /tmp/conda-tmp/environment.yml; \
diff --git a/.devcontainer/environment.yml b/.devcontainer/environment.yml
index 59cc01f..6e30dbb 100644
--- a/.devcontainer/environment.yml
+++ b/.devcontainer/environment.yml
@@ -3,12 +3,13 @@ channels:
   - conda-forge
   - defaults
 dependencies:
+  - numpy
+  - cupy
+  - jax[cuda]
+  - scipy
   - pre-commit==3.7.1
   - black==24.4.2
   - flake8==7.1.0
-  - numpy==2.0.0
-  - scipy==1.14.0
-  - jax[cuda]==0.4.28
   - parameterized==0.9.0
   - pytest==8.2.2
   - pytest-rerunfailures==14.0