docker19.03+调用宿主机nvidia显卡,配置默认runtime=nvidia

若是想在容器里调用宿主机的nvidia显卡加速运算,docker 19.03+版本摒弃了以前的nvidia-docker2的实现方式,docker 运行镜像时候只需在run 后加--gpus all的参数,可是在docker-compose里确没有--gpus=all或者runtime=nvidia的参数让docker调用nvidia显卡,以前还觉得了nvidia-docker官方没有跟上dock
相关文章
相关标签/搜索