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