add cupy dependency

cai4cai · Oct 21, 2024 · cc66ff9 · cc66ff9
1 parent 45c117b
commit cc66ff9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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