docker踩坑记录二

昨天安装了docker后，需要启动tf镜像命令

docker run --runtime=nvidia -it tensorflow/tensorflow:1.5.0-gpu-py3 bash

然后就一直报错：

requirement error: unsatisfied condition: cuda >= 9.0\\\\n\\\"\"": unknown.

明明已经装好了cuda9.0, 也配置了环境变量，但是还是不行，后来看到https://github.com/NVIDIA/nvidia-docker/issues/543说

You need to upgrade your **driver**. This is 9.0, not 8.0.
Or, you can use the `nvidia/cuda:8.0-devel` image and keep your driver.

需要更新英伟达的驱动。

检查了一下，当前机器的驱动是374版本的 （可以用nvida-smi 命令查看驱动版本），看了一下官网说明，9.0需要装384版本的驱动

https://github.com/NVIDIA/nvidia-docker/wiki/CUDA#requirements

CUDA toolkit version	Driver version	GPU architecture
6.5	>= 340.29	>= 2.0 (Fermi)
7.0	>= 346.46	>= 2.0 (Fermi)
7.5	>= 352.39	>= 2.0 (Fermi)
8.0	== 361.93 or >= 375.51	== 6.0 (P100)
8.0	>= 367.48	>= 2.0 (Fermi)
9.0	>= 384.81	>= 3.0 (Kepler)
9.1	>= 387.26	>= 3.0 (Kepler)
9.2	>= 396.26	>= 3.0 (Kepler)
10.0	>= 384.130, < 385.00	Tesla GPUs
10.0	>= 410.48	>= 3.0 (Kepler)

 

于是重新去英伟达网站上下载驱动安装

https://www.nvidia.com/content/DriverDownload-March2009/confirmation.php?url=/tesla/384.145/NVIDIA-Linux-x86_64-384.145.run&lang=us&type=Tesla

找到对应自己卡型号的驱动，装一下就行了。

./NVIDIA-Linux-x86_64-384.145.run

然后再执行，OK，不报错了！

root@4ecc101654e9:/notebooks#