Skip to content

Commit

Permalink
hub image: pin latest conda-forge available cudatoolkit version
Browse files Browse the repository at this point in the history
  • Loading branch information
consideRatio committed Sep 2, 2021
1 parent 2d991ce commit b77bbb9
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions hub.jupytearth.org-image/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -207,11 +207,15 @@ RUN echo "Installing conda packages..." \
# I think the driver will be updatable by editing a DaemonSet installed
# in the k8s cluster automatically by eksctl.
#
# If the wanted cudatoolkit version is know, then what driver should be
# used? Answer: https://docs.cupy.dev/en/stable/install.html#installation
# https://docs.cupy.dev/en/stable/install.html#installation
# https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
#
# Check latest cudatoolkit version on conda-forge, currently 11.2.2:
# - https://anaconda.org/conda-forge/cudatoolkit
# - https://github.com/conda-forge/cudatoolkit-feedstock
#
cupy \
cudatoolkit=11.4 \
cudatoolkit=11.2.2 \
#
# temp workarounds
&& echo "Installing conda packages complete!"
Expand Down

0 comments on commit b77bbb9

Please sign in to comment.