達(dá)到以下條件即可：
1 docker內(nèi)部和宿主機(jī)的顯卡驅(qū)動要一致
說白了就是你的cuda、cudnn基于那個顯卡驅(qū)動，你就要用哪個，內(nèi)外都要部署
我測試過內(nèi)部不部署，運行會失敗

2 內(nèi)部雖然裝了顯卡驅(qū)動，但是沒有實體卡，所以還要做一步設(shè)備映射
就是把外部的GPU設(shè)備，掛在到docker的dev下
運行時使用–device選項掛在設(shè)備

總結(jié)

以上是生活随笔為你收集整理的ubuntu下docker使用GPU的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。