达到以下条件即可：
1 docker内部和宿主机的显卡驱动要一致
说白了就是你的cuda、cudnn基于那个显卡驱动，你就要用哪个，内外都要部署
我测试过内部不部署，运行会失败
2 内部虽然装了显卡驱动，但是没有实体卡，所以还要做一步设备映射
就是把外部的GPU设备，挂在到docker的dev下
运行时使用–device选项挂在设备
来源：CSDN
作者：凌风探梅
链接：https://blog.csdn.net/Real_Myth/article/details/82963703