move nvidia-smi

QuantEcon · Jan 31, 2024 · 2d71048 · 2d71048
1 parent 4d4b35a
commit 2d71048
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuda-12.3.1-anaconda-2023-09-py311/Dockerfile b/cuda-12.3.1-anaconda-2023-09-py311/Dockerfile
@@ -41,9 +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_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+RUN nvidia-smi
 COPY check_jax.py .
 RUN python check_jax.py