From 66b5175f11c1f9b52e410b4de54b8cdb63d60cb2 Mon Sep 17 00:00:00 2001 From: mmcky Date: Thu, 1 Feb 2024 09:56:33 +1100 Subject: [PATCH] try _pip config for install and show nvidia-smi --- cuda-12.3.1-anaconda-2023-09-py311/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cuda-12.3.1-anaconda-2023-09-py311/Dockerfile b/cuda-12.3.1-anaconda-2023-09-py311/Dockerfile index 44490fc..e60988a 100644 --- a/cuda-12.3.1-anaconda-2023-09-py311/Dockerfile +++ b/cuda-12.3.1-anaconda-2023-09-py311/Dockerfile @@ -41,8 +41,9 @@ RUN echo "source activate quantecon" > ~/.bashrc ENV PATH /opt/conda/envs/quantecon/bin:$PATH # Install JAX +RUN nvidia-smi RUN nvcc --version RUN echo "$LD_LIBRARY_PATH" -RUN pip install --upgrade "jax[cuda12_local]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html +RUN pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html COPY check_jax.py . RUN python check_jax.py \ No newline at end of file