docker容器使用GPU方法實(shí)現(xiàn)
為了使docker容器可以使用nvidia-docker,可以有如下幾種辦法：
1、無nvidia-docker
在早期的時(shí)候，沒有nvidia-docker，可以通過在容器內(nèi)再部署一遍nvidia GPU驅(qū)動(dòng)解決。同理，其他設(shè)備如果想在容器里使用，也可以采用在容器里重新安裝一遍驅(qū)動(dòng)解決。
2、nvidia-docker1.0
nvidia-docker是英偉達(dá)公司專門用來為docker容器使用nvidia GPU而設(shè)計(jì)的，設(shè)計(jì)方案就是把宿主機(jī)的GPU驅(qū)動(dòng)文件映射到容器內(nèi)部使用，可以通過tensorflow生成GPU驅(qū)動(dòng)文件夾。
3、nvidia-docker2.0
nvidia-docker2.0對(duì)nvidia-docker1.0進(jìn)行了很大的優(yōu)化，不用再映射宿主機(jī)GPU驅(qū)動(dòng)了，直接把宿主機(jī)的GPU運(yùn)行時(shí)映射到容器即可。啟動(dòng)方式示例：
nvidia-docker run -d -e NVIDIA_VISIBLE_DEVICES=all --name nvidia_docker_test nvidia/cuda:10.0-base /bin/sh -c "while true; do echo hello world; sleep 1; done"
4、安裝docker-19.03及以上版本
docker19.03及以上版本，已經(jīng)內(nèi)置了nvidia-docker，無需再單獨(dú)部署nvidia-docker了。安裝方式如下：
安裝docker：
yum install -y yum-utils yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo yum-config-manager --enable docker-ce-nightly yum-config-manager --enable docker-ce-test yum install docker-ce docker-ce-cli containerd.io systemctl start docker
安裝nvidia-container-toolkit
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo sudo yum install -y nvidia-container-toolkit sudo systemctl restart docker
啟動(dòng)容器：
docker run --gpus all nvidia/cuda:10.0-base /bin/sh -c "while true; do echo hello world; sleep 1; done"
進(jìn)入容器并輸入nvidia-smi驗(yàn)證。
到此這篇關(guān)于docker容器使用GPU方法實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)docker使用GPU內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了CentOS7中配置Docker的yum源并安裝使用詳解，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-09-09
這篇文章主要介紹了Docker容器無法解析域名的處理問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,2023-11-11
本文主要使用三種不同的方式將數(shù)據(jù)從Docker主機(jī)掛載到容器中，分別為卷（volumes），綁定掛載（bind?mounts），臨時(shí)文件系tmpfs，還介紹了數(shù)據(jù)卷容器、數(shù)據(jù)卷的備份與恢復(fù)，對(duì)Docker?存儲(chǔ)管理相關(guān)知識(shí)感興趣的朋友一起看看吧2022-07-07
docker-maven-plugin 插件無法拉取對(duì)應(yīng)jar包問題
這篇文章主要介紹了docker-maven-plugin 插件無法拉取問題,總是報(bào)錯(cuò)，如何解決這個(gè)問題呢，下面小編給大家?guī)砹私鉀Q方法，一起看看吧2021-09-09
這篇文章主要介紹了Docker 容器操作指令匯總詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-10-10
使用docker快速部署Nginx、Redis、MySQL、Tomcat及制作鏡像的方法
這篇文章介紹了如何使用Docker快速部署Nginx、Redis、MySQL、Tomcat以及制作鏡像的過程,它涵蓋了從基礎(chǔ)鏡像的使用、Dockerfile的編寫到應(yīng)用的構(gòu)建和部署的詳細(xì)步驟,通過本文,可以掌握如何利用Docker簡化應(yīng)用的部署和管理,感興趣的朋友跟隨小編一起看看吧2024-11-11
這篇文章主要介紹了Docker端口占用問題分析以及解決方案,主要步驟包括檢查端口占用、處理Windows環(huán)境下的特殊問題、解決權(quán)限問題、修改端口映射和考慮其他可能原因,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2025-04-04
使用docker啟動(dòng)redis守護(hù)進(jìn)程的詳細(xì)流程
在現(xiàn)代軟件開發(fā)中,緩存和消息隊(duì)列是提高應(yīng)用性能和擴(kuò)展性的關(guān)鍵,Redis 是一種流行的內(nèi)存數(shù)據(jù)結(jié)構(gòu)存儲(chǔ)系統(tǒng),使用 Docker 容器化技術(shù)可以方便地部署和管理 Redis 服務(wù),本文給大家介紹了使用docker啟動(dòng)redis守護(hù)進(jìn)程的詳細(xì)流程,需要的朋友可以參考下2025-01-01
IDEA直接打包Docker鏡像的實(shí)現(xiàn)
Docker提供了多種方法來打包鏡像,本文主要介紹了IDEA直接打包Docker鏡像的實(shí)現(xiàn),具有一定的參考價(jià)值,感興趣的可以了解一下2024-06-06
今天小編就為大家分享一篇關(guān)于使用OpenSSL生成Kubernetes證書的介紹，小編覺得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來看看吧2019-01-01