Linux安装Nvidia Docker
注意: 由于程序运行于容器之中,所以镜像中一般都是带有CUDA、CUDNN库的,因此只需要在docker所在的主机上安装显卡驱动即可,无需费太大力气去安装cuda、cuddn之类的东西。
注意: 由于程序运行于容器之中,所以镜像中一般都是带有CUDA、CUDNN库的,因此只需要在docker所在的主机上安装显卡驱动即可,无需费太大力气去安装cuda、cuddn之类的东西。
docker调用GPU时可以明确指定用哪一张显卡,同时也可以指定数量

Ubuntu16.04
Nvidia 384
CUDA 8.0
cuDNN 5
TensorFlow、CUDA、cuDNN的版本关系我时常懵懵的,经常出现各种不支持的情况,比如今天安装了TensorFlow1.10.1,报错:ImportError: libcudnn.so.7: cannot open shared object file: No such file or dictionary
此文档来源于Jermine的个人blog : https://jermine.vdo.pub/linux/ubuntu-16.04-reinstall-cuda/
注意 : 由于tensorflow的GPU版本依赖nvidia的cuda、cudnn库,因此一般需要包含cuda和cudnn的链接库文件,普遍做法是通过主机安装cudnn、cuda的方式。这里还有另外两种方式可以选择: