Update Dockerfile

yufree · Aug 28, 2024 · 4362022 · 4362022
1 parent 1c0aeed
commit 4362022
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspace/Dockerfile b/deepspace/Dockerfile
@@ -19,7 +19,7 @@ RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -
 ENV PATH=/opt/conda/bin:$PATH
 
 RUN conda create -n gpu python=3.12 -y \
-    && conda install -n gpu -c conda-forge jupyter ipykernel tables h5py numpy pandas scipy scikit-learn joblib matplotlib cupy-cuda12x \
+    && conda install -n gpu -c conda-forge jupyter ipykernel tables h5py numpy pandas scipy scikit-learn joblib matplotlib cupy \
     && conda clean -afy
 
 RUN wget -qO- https://cloud.r-project.org/bin/linux/ubuntu/marutter_pubkey.asc | tee -a /etc/apt/trusted.gpg.d/cran_ubuntu_key.asc \