安裝NVIDIA docker plugin

GPU云主機(jī)集成CUDA & NVIDIA DOCKER鏡像方案

一、預(yù)安裝前置條件
1、系統(tǒng)和內(nèi)核版本支持
2、 CUDA的GPU支持
二、NVIDIA驅(qū)動(dòng)環(huán)境安裝

1、安裝CUDA Toolkit
2、設(shè)置cuda環(huán)境變量
3、安裝 docker-18.06.1
4、安裝NVIDIA docker plugin
三、 驅(qū)動(dòng)環(huán)境檢視
1、 cuda驅(qū)動(dòng)環(huán)境檢視
2、 cuda設(shè)備用例查詢
3、docker插件安裝環(huán)境檢視
4、NVIDIA docker plugin用例測(cè)試
四、NGC測(cè)試
1、 運(yùn)行tensorflow容器
2、用例功能測(cè)試
五、GPU云主機(jī)產(chǎn)品新增CUDA和NVIDIA-DOCKER自定義鏡像功能
1、產(chǎn)品交互形態(tài)
2、CUDA自定義鏡像制作方案
3、NVIDIA DOCKER自定義鏡像制作方案

一、 預(yù)安裝前置條件
1、 系統(tǒng)和內(nèi)核版本支持

cat /etc/*release

uname -a

2、CUDA的GPU支持

lspci | grep -i nvidia

二、 環(huán)境集成安裝
1、安裝CUDA Toolkit
rpm -i cuda-repo-rhel7-9-1-local-9.1.85-1.x86_64.rpm
yum clean all
yum install cuda

2、設(shè)置cuda環(huán)境變量
export PATH=$PATH:/usr/local/cuda-9.1/bin
export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64

3、安裝 docker-18.06.1
①安裝
yum install -y docker-ce-selinux-17.03.2.ce-1.el7.centos.noarch.rpm
yum install -y docker-ce-18.06.1.ce-3.el7.x86_64.rpm

②、docker 配置文件修改

③、啟動(dòng)docker
system restart docker

4、安裝NVIDIA docker-plugin
rpm -i libseccomp-2.3.1-3.el7.x86_64.rpm
rpm -i libnvidia-container1-1.0.0-0.1.rc.2.x86_64.rpm
rpm -i libnvidia-container-tools-1.0.0-0.1.rc.2.x86_64.rpm
rpm -i nvidia-container-runtime-hook-1.4.0-2.x86_64.rpm
rpm -i nvidia-container-runtime-2.0.0-1.docker18.06.1.x86_64.rpm
rpm -i nvidia-docker2-2.0.3-1.docker18.06.1.ce.noarch.rpm

三、 驅(qū)動(dòng)環(huán)境檢視
1、 cuda驅(qū)動(dòng)驗(yàn)證
nvidia-smi
nvcc --version

2、 cuda設(shè)備用例查詢
cd /usr/local/cuda-9.1/samples/1_Utilities/deviceQuery
make
./deviceQuery

3、 nvidia docker plugin安裝驗(yàn)證
nvidia-docker version

4、 docker用例測(cè)試
docker run hello-world

四、NGC測(cè)試
1、運(yùn)行tensorflow容器：
docker run --runtime=nvidia -it --rm -v
/tmp/gpu-cuda-install/benchmarks-cnn_tf_v1.8_compatible:/workspace
nvcr.io/nvidia/tensorflow:18.02-py2 bash

2、用例功能測(cè)試：
cd /opt/tensorflow/nvidia-examples/cnn && python nvcnn.py --model=resnet50
--batch_size=64 --num_gpus=1 --fp16

五、GPU云主機(jī)產(chǎn)品新增CUDA和NVIDIA-DOCKER鏡像功能
1、產(chǎn)品交互形態(tài)
操作系統(tǒng)Centos7.2-64選項(xiàng)子選項(xiàng)框新增cuda鏡像和nvidia-docker鏡像

2、 CUDA自定義鏡像制作方案

3、NVIDIA DOCKER自定義鏡像制作方案
END

版權(quán)聲明：本文內(nèi)容由互聯(lián)網(wǎng)用戶自發(fā)貢獻(xiàn)，版權(quán)歸作者所有，本社區(qū)不擁有所有權(quán)，也不承擔(dān)相關(guān)法律責(zé)任。如果您發(fā)現(xiàn)本社區(qū)中有涉嫌抄襲的內(nèi)容，歡迎發(fā)送郵件至：[email protected]
進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，本社區(qū)將立刻刪除涉嫌侵權(quán)內(nèi)容。

• ? 上一篇：【程序人生】2018與2019年終總結(jié)——悟已往之不諫，知來(lái)者之可追
• ? 下一篇：微軟到底有多恐怖?