Skip to content

Commit

Permalink
remove nvidia-smi
Browse files Browse the repository at this point in the history
  • Loading branch information
mmcky committed Jan 31, 2024
1 parent 2d71048 commit 545cde1
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion cuda-12.3.1-anaconda-2023-09-py311/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,5 @@ ENV PATH /opt/conda/envs/quantecon/bin:$PATH
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

0 comments on commit 545cde1

Please sign in to comment.