Skip to content

Commit

Permalink
⬆️ Bump cuda driver from 390.46 to 410.73
Browse files Browse the repository at this point in the history
The Kahutea HPC server I've been using has just updated its CUDA drivers to version 410.79, so I'm matching that (almost). Note that we are sticking to cudatoolkit 9.0 instead of 10.0 and cudnn 7.1.2 instead of 7.2.1 (or higher) as that is what the tensorflow-gpu binaries on pypi are pre-compiled with. It is possible to use an older cudatoolkit on a newer driver (as we are doing now), but not vice versa. I.e. cudatoolkit 9.0 works on a >=410.48 driver, but cudatoolkit 10.0 won't work on a 396.xx version driver. Basically, just get the latest driver!!

Also made a slight adjustment to Dockerfile script. Add the conda-forge/label/dev channel system-wide for consistency.
  • Loading branch information
weiji14 committed Jan 9, 2019
1 parent e1e7144 commit 4dd92c8
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ RUN cd /tmp && \
echo "e1045ee415162f944b6aebfe560b8fee *Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh" | md5sum -c - && \
/bin/bash Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh -f -b -p $CONDA_DIR && \
rm Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh && \
$CONDA_DIR/bin/conda config --prepend channels conda-forge/label/dev && \
$CONDA_DIR/bin/conda config --system --prepend channels conda-forge/label/dev && \
$CONDA_DIR/bin/conda config --system --prepend channels conda-forge && \
$CONDA_DIR/bin/conda config --system --set auto_update_conda false && \
$CONDA_DIR/bin/conda config --system --set show_channel_urls true && \
Expand Down
2 changes: 1 addition & 1 deletion environment.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ channels:
- nodefaults
dependencies:
- defaults::cudnn=7.1.2[md5=4a402b88bb77e6ab2dcf3bfe6522f9cf]
- hcc::cuda_driver=390.46[md5=8fb0b6c39a9bf6128b1191db53ed903e]
- hcc::cuda_driver=410.73[md5=941787b750b372f4a240287634589d24]
- defaults::cudatoolkit=9.0[md5=5d0febed868b80a18e74077d5d0f17bc]
- gmt=6.0.0a17[md5=bea1e9a2cc29280f8ba173123f115496]
- pip=18.1[md5=d68c7e5109ba0bf4b1cfe60f0f47870a]
Expand Down

0 comments on commit 4dd92c8

Please sign in to comment.