ubuntu下docker使用GPU

版权声明:转载注明出处 http://blog.csdn.net/Real_Myth,如果发现有未完成的文章,请及时留言告知 https://blog.csdn.net/Real_Myth/article/details/82963703

达到以下条件即可:
1 docker内部和宿主机的显卡驱动要一致
说白了就是你的cuda、cudnn基于那个显卡驱动,你就要用哪个,内外都要部署
我测试过内部不部署,运行会失败

2 内部虽然装了显卡驱动,但是没有实体卡,所以还要做一步设备映射
就是把外部的GPU设备,挂在到docker的dev下
运行时使用–device选项挂在设备

猜你喜欢

转载自blog.csdn.net/Real_Myth/article/details/82963703