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

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。