Ubuntu 18.04 Server 搭建tensorflow docker運行環境
- 2019 年 10 月 5 日
- 筆記
一、系統安裝
二、顯示卡驅動安裝
顯示卡型號：Nvidia GTX1050.
1、首先去Nvidia官網下載驅動：
image
找到自己的顯示卡，選擇Linux版本進行下載，下載的是.run文件。
2、禁用nouveau
安裝NVIDIA需要把系統自帶的驅動禁用，打開文件：
sudo gedit /etc/modprobe.d/blacklist.conf
在文本最後添加以下內容：
blacklist nouveau option nouveau modeset=0
命令窗口會提示warn，無視之。
保存退出，執行以下命令生效：
sudo update-initramfs -u
重啟電腦後輸入：
lsmod | grep nouveau
沒有任何輸出說明禁用成功。
3、安裝NVIDIA顯示卡驅動
在安裝驅動程式的過程中，會因為缺少gcc、g++、make等development tool而報錯導致無法完成驅動程式安裝（ERROR：Ubable to find the development tool 'make' in your path…），不用擔心，手動安裝這些開發包後再次執行安裝指令即可。另外，安裝過程中出現的彈框根據默認選項選擇即可。手動安裝development tool指令為:
sudo apt-get install gcc sudo apt-get install g++ sudo apt-get install make
4、 查看顯示卡驅動版本
可以通過nvidia-smi命令查看GPU和驅動程式資訊，若出現以下結果，則表明驅動程式安裝成功。
安裝時沒有截圖，所以用了網上的圖片
三、安裝docker
1、安裝基礎docker服務
安裝基礎的docker請參考Ubuntu下通過docker部署springboot項目 安裝和配置docker部分。
2、安裝nvidia-docker2服務
想要在docker容器中使用宿主機的gpu，則需要安裝一個nvidia-docker的插件。下面介紹安裝方法： 第一種：
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb sudo dpkg -i /tmp/nvidia-docker*.deb
第二種：
sudo apt-get install nvidia-docker2
本人通過第二種方式安裝，第一種方式未經測試。 可以直接去nvidia-docker 的github看看：https://github.com/NVIDIA/nvidia-docker/
3、測試nvidia-docker
nvidia-docker run --rm nvidia/cuda nvidia-smi
然後在執行下面這句，默認用nvdia-docker替代docker命令:
echo 'alias docker=nvidia-docker' >> ~/.bashrc bash
最後就可以下載tensorflow鏡像進行測試了，這裡就不詳細介紹了。 附贈 tensorflow docker hub 地址
