摘要: 如果要在容器内跑使用GPU的程序,需要安装nvidia-docker,这样可以在docker内直接调用宿主机的GPU。 1. 如果直接在容器内安装nvidia的驱动,配置cuda环境,在使用时将需要指定宿主机的GPU位置,类似于/devices/gpu0 这样的。 nvidia-docker 安装方 阅读全文
posted @ 2024-03-19 17:27 谷小雨 阅读(617) 评论(0) 推荐(0)