在Ubuntu系統(tǒng)上,為Docker安裝NVIDIA運行時環(huán)境(NVIDIA Container Runtime for Docker)通常涉及以下步驟:
-
安裝Docker(如果您還沒有安裝):
首先,確保您的系統(tǒng)已經(jīng)安裝了Docker。您可以使用以下命令進行安裝:
sudo apt update sudo apt install docker.io
-
添加NVIDIA容器運行時的存儲庫:
執(zhí)行以下命令以添加NVIDIA的Apt存儲庫:
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
-
安裝NVIDIA容器運行時:
更新Apt包列表并安裝NVIDIA容器運行時:
sudo apt update sudo apt install nvidia-container-runtime
-
配置Docker以使用NVIDIA運行時:
編輯或創(chuàng)建Docker的配置文件(如果不存在)
/etc/docker/daemon.json
,并添加以下內容:{ "default-runtime": "nvidia", "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } } }
保存并關閉文件。
-
重新啟動Docker服務:
sudo systemctl restart docker
-
驗證NVIDIA運行時的安裝:
使用以下命令運行一個測試容器來驗證NVIDIA運行時的安裝:
docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
如果一切正常,您應該能夠看到NVIDIA的
nvidia-smi
命令的輸出,顯示了GPU的詳細信息。文章來源:http://www.zghlxwxcb.cn/news/detail-821803.html
這樣,您的Ubuntu系統(tǒng)上的Docker就已配置為使用NVIDIA容器運行時了。確保您的NVIDIA顯卡驅動和CUDA工具包也已正確安裝,以便與Docker容器一起使用。文章來源地址http://www.zghlxwxcb.cn/news/detail-821803.html
到了這里,關于【Ubuntu】為Docker安裝NVIDIA運行時環(huán)境(NVIDIA Container Runtime for Docker)的文章就介紹完了。如果您還想了解更多內容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!