如果要在容器內跑使用GPU的程式,需要安裝nvidia-docker,這樣可以在docker內直接呼叫宿主機的GPU。
1. 如果直接在容器內安裝nvidia的驅動,配置cuda環境,在使用時將需要指定宿主機的GPU位置,類似於/devices/gpu0 這樣的。
nvidia-docker 安裝方法按照 nvidia官方文件即可;部落格可 參考
驗證可用後,可以去 docker官方 找 nvidia/cuda 相關的映象,這些映象像其它映象一樣用即可,裡面已經安裝了對應版本的cuda等內容。
2. 版本含義,如:10.1-cudnn7-runtime-ubuntu18.04表示:ubuntu系統庫,cuda10.1,cudnn7,runtime版。
** 三個版本的nvidia-docker(三個版本的size依次增大) ——參考
base版:只有基礎的庫(libcudart);
runtime版:有所有的共享庫,呼叫即可;
devel版:即開發版,包含共享庫和編譯、除錯工具,標頭檔案等,有bin/目錄和nvcc,libcudnn在 /usr/lib/x86_64-linux-gnu/libcudnn.so。