之前按照一個博客安裝 運行還是各種坑
按照如下的步奏成功了
安裝(2.0版)
Felix Abecassis編輯了此頁面 on 1 Dec 2017 · 5次修訂
先決條件
運行nvidia-docker 2.0的先決條件列表如下所述。
有關如何爲Linux發行版安裝Docker的信息,請參閱Docker文檔。
- 內核版本> 3.10的GNU / Linux x86_64
- Docker> = 1.12
- 採用架構的NVIDIA GPU> Fermi(2.1)
- NVIDIA驅動程序〜= 361.93(舊版本未經測試)
您的驅動程序版本可能會限制您的CUDA功能(請參閱CUDA要求)
刪除nvidia-docker 1.0
在繼續之前,必須徹底刪除nvidia-docker軟件包的1.0版。
您必須停止並刪除所有使用nvidia-docker 1.0啓動的容器。
Ubuntu發行版
docker volume ls -q -f driver = nvidia-docker | xargs -r -I {} -n1 docker ps -q -a -f volume = {} | xargs -r docker rm -f
sudo apt-get purge nvidia-docker
CentOS發行版
docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo yum remove nvidia-docker
安裝2.0版
確保已爲您的發行版安裝了NVIDIA驅動程序和受支持的Docker 版本(請參閱先決條件)。
如果您有自定義/etc/docker/daemon.json
,則nvidia-docker2
程序包可能會覆蓋它。
Ubuntu發行版
- 按照此處的說明爲您的分發安裝存儲庫。
- 安裝
nvidia-docker2
軟件包並重新加載Docker守護程序配置:
sudo apt-get install nvidia-docker2
sudo pkill -SIGHUP dockerd
CentOS發行版
- 按照此處的說明爲您的分發安裝存儲庫。
- 安裝
nvidia-docker2
軟件包並重新加載Docker守護程序配置:
sudo yum install nvidia-docker2
sudo pkill -SIGHUP dockerd
舊版本的Docker
您必須固定兩者的版本nvidia-docker2
以及nvidia-container-runtime
安裝時,例如:
sudo apt-get install -y nvidia-docker2=2.0.1+docker1.12.6-1 nvidia-container-runtime=1.1.0+docker1.12.6-1
使用apt-cache madison nvidia-docker2 nvidia-container-runtime
或yum search --showduplicates nvidia-docker2 nvidia-container-runtime
列出可用版本。
其他發行版和體系結構
按照此處的說明爲您的分發安裝存儲庫。
基本用法
nvidia-docker向Docker守護程序註冊一個新的容器運行時。使用時
必須選擇nvidia
運行時docker run
:
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
運行成功截圖
之後就可以進行下一步