Skip to content

Commit 4dd92c8

Browse files
committed
⬆️ Bump cuda driver from 390.46 to 410.73
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.
1 parent e1e7144 commit 4dd92c8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Dockerfile

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ RUN cd /tmp && \
2424
echo "e1045ee415162f944b6aebfe560b8fee *Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh" | md5sum -c - && \
2525
/bin/bash Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh -f -b -p $CONDA_DIR && \
2626
rm Miniconda3-${MINICONDA_VERSION}-Linux-x86_64.sh && \
27-
$CONDA_DIR/bin/conda config --prepend channels conda-forge/label/dev && \
27+
$CONDA_DIR/bin/conda config --system --prepend channels conda-forge/label/dev && \
2828
$CONDA_DIR/bin/conda config --system --prepend channels conda-forge && \
2929
$CONDA_DIR/bin/conda config --system --set auto_update_conda false && \
3030
$CONDA_DIR/bin/conda config --system --set show_channel_urls true && \

environment.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ channels:
55
- nodefaults
66
dependencies:
77
- defaults::cudnn=7.1.2[md5=4a402b88bb77e6ab2dcf3bfe6522f9cf]
8-
- hcc::cuda_driver=390.46[md5=8fb0b6c39a9bf6128b1191db53ed903e]
8+
- hcc::cuda_driver=410.73[md5=941787b750b372f4a240287634589d24]
99
- defaults::cudatoolkit=9.0[md5=5d0febed868b80a18e74077d5d0f17bc]
1010
- gmt=6.0.0a17[md5=bea1e9a2cc29280f8ba173123f115496]
1111
- pip=18.1[md5=d68c7e5109ba0bf4b1cfe60f0f47870a]

0 commit comments

Comments
 (0)