新聞中心
Docker 19.03版本提供了對(duì)NVIDIA顯卡的支持,使得在Docker容器中使用GPU變得更加方便,本文將詳細(xì)介紹如何在Docker 19.03中使用NVIDIA顯卡。

創(chuàng)新互聯(lián)是專業(yè)的南湖網(wǎng)站建設(shè)公司,南湖接單;提供成都網(wǎng)站設(shè)計(jì)、成都網(wǎng)站建設(shè),網(wǎng)頁(yè)設(shè)計(jì),網(wǎng)站設(shè)計(jì),建網(wǎng)站,PHP網(wǎng)站建設(shè)等專業(yè)做網(wǎng)站服務(wù);采用PHP框架,可快速的進(jìn)行南湖網(wǎng)站開發(fā)網(wǎng)頁(yè)制作和功能擴(kuò)展;專業(yè)做搜索引擎喜愛的網(wǎng)站,專業(yè)的做網(wǎng)站團(tuán)隊(duì),希望更多企業(yè)前來(lái)合作!
你需要確保你的系統(tǒng)已經(jīng)安裝了NVIDIA顯卡驅(qū)動(dòng),并且Docker服務(wù)也已經(jīng)啟動(dòng),如果你的系統(tǒng)中沒有NVIDIA顯卡,那么你可能無(wú)法使用Docker的GPU支持。
接下來(lái),你需要安裝NVIDIA Container Toolkit,這是一個(gè)用于在Docker容器中使用GPU的工具包,你可以在NVIDIA的官方網(wǎng)站上下載,安裝完成后,你需要重啟Docker服務(wù)。
在安裝完NVIDIA Container Toolkit后,你就可以在Docker中使用GPU了,你可以通過(guò)以下命令來(lái)檢查是否已經(jīng)成功啟用了GPU支持:
docker run --rm nvidia/cuda:10.2-base nvidia-smi
這個(gè)命令會(huì)運(yùn)行一個(gè)包含CUDA 10.2的基礎(chǔ)鏡像的Docker容器,并顯示關(guān)于GPU的信息,如果你看到了關(guān)于GPU的信息,那么就說(shuō)明你已經(jīng)成功啟用了GPU支持。
在使用Docker時(shí),你可以通過(guò)在運(yùn)行容器時(shí)添加`–gpus all`參數(shù)來(lái)啟用所有可用的GPU。
docker run --gpus all -it ubuntu nvidia-smi
這個(gè)命令會(huì)運(yùn)行一個(gè)Ubuntu容器,并使用所有可用的GPU,在這個(gè)容器中,你可以運(yùn)行任何需要GPU的程序。
除了使用`–gpus all`參數(shù)外,你還可以使用`–gpus `參數(shù)來(lái)指定使用特定數(shù)量的GPU,如果你想使用兩個(gè)GPU,你可以運(yùn)行:
docker run --gpus '"2"' -it ubuntu nvidia-smi
這個(gè)命令會(huì)運(yùn)行一個(gè)Ubuntu容器,并使用兩個(gè)GPU,在這個(gè)容器中,你可以運(yùn)行需要GPU的程序。
Docker 19.03提供了對(duì)NVIDIA顯卡的支持,使得在Docker容器中使用GPU變得更加方便,你只需要安裝NVIDIA Container Toolkit,并在運(yùn)行容器時(shí)添加相應(yīng)的參數(shù),就可以在Docker中使用GPU了。
相關(guān)問(wèn)題與解答**
問(wèn)題1:我安裝了NVIDIA Container Toolkit,但是仍然無(wú)法在Docker中使用GPU,怎么辦?
答:請(qǐng)檢查你是否已經(jīng)正確安裝了NVIDIA Container Toolkit,以及是否正確重啟了Docker服務(wù),如果問(wèn)題仍然存在,你可以嘗試卸載并重新安裝NVIDIA Container Toolkit。
問(wèn)題2:我在運(yùn)行Docker容器時(shí)使用了`–gpus all`參數(shù),但是我看到的GPU信息和預(yù)期的不同,為什么?
答:`–gpus all`參數(shù)會(huì)啟用所有可用的GPU,如果你只看到一部分GPU的信息,可能是因?yàn)槠渌鸊PU沒有被正確識(shí)別或者已經(jīng)被其他程序占用,你可以嘗試運(yùn)行`nvidia-smi`命令來(lái)查看所有的GPU信息。
問(wèn)題3:我在運(yùn)行Docker容器時(shí)使用了`–gpus `參數(shù),但是我指定的GPU數(shù)量和我實(shí)際使用的GPU數(shù)量不一致,為什么?
答:`–gpus `參數(shù)只是指定了你希望Docker使用多少個(gè)GPU,實(shí)際上,Docker可能會(huì)根據(jù)你的硬件配置和使用情況動(dòng)態(tài)調(diào)整使用的GPU數(shù)量,即使你指定了特定的GPU數(shù)量,實(shí)際使用的GPU數(shù)量也可能會(huì)有所不同。
當(dāng)前名稱:dockernvidia
URL標(biāo)題:http://www.dlmjj.cn/article/cdsgsod.html


咨詢
建站咨詢
