From 4dd92c8f29b007619ca905a338ffb7050f5157ec Mon Sep 17 00:00:00 2001 From: Wei Ji Date: Wed, 9 Jan 2019 15:44:52 +1300 Subject: [PATCH] :arrow_up: 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. --- Dockerfile | 2 +- environment.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index f407289..ad66173 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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 && \ diff --git a/environment.yml b/environment.yml index edfe329..bffa4ca 100644 --- a/environment.yml +++ b/environment.yml @@ -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]