达到以下条件即可:
1 docker内部和宿主机的显卡驱动要一致
说白了就是你的cuda、cudnn基于那个显卡驱动,你就要用哪个,内外都要部署
我测试过内部不部署,运行会失败
2 内部虽然装了显卡驱动,但是没有实体卡,所以还要做一步设备映射
就是把外部的GPU设备,挂在到docker的dev下
运行时使用–device选项挂在设备
达到以下条件即可:
1 docker内部和宿主机的显卡驱动要一致
说白了就是你的cuda、cudnn基于那个显卡驱动,你就要用哪个,内外都要部署
我测试过内部不部署,运行会失败
2 内部虽然装了显卡驱动,但是没有实体卡,所以还要做一步设备映射
就是把外部的GPU设备,挂在到docker的dev下
运行时使用–device选项挂在设备