基于Docker的tensorflow:cpu计算

版权声明： https://blog.csdn.net/u013531940/article/details/79705581

环境配置：一台服务器（带GPU）

一、安装GPU驱动

1. 禁用ubuntu的GPU驱动（如果在安装驱动过程中，报需禁用nouveau的错误，则按照第一步操作，否则直接第二步）

    卸载已存在的NVIDIA驱动版本

apt-get remove --purge nvidia*

    禁用 nouveau

root@yujmo:~# vim /etc/modprobe.d/blacklist.conf #在文末加上下图中的内容

update-initramfs -u

reboot

2. 安装GPU驱动（官方）

root@yujmo:~# ./NVIDIA-Linux-x86_64-390.42.run 

二、安装nvidia-docker

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list

apt-get update

apt-get install -y nvidia-docker2

pkill -SIGHUP dockerd

三、测试


docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

root@yujmo:~# nvidia-docker run --rm nvidia/cuda nvidia-smi