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