docker 启动参数上加 -e NVIDIA_VISIBLE_DEVICES=1(0代表第一块显卡,1代表第二块显卡,不加参数是全部启动)

docker run -it  --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=1 需要启动的docker名称 bash

多显卡机器上,如果显卡0正在被其他程序占用,但是程序指定显卡0,你就可以加这个命令,只启动部分显卡,docker 里面看到cuda 0编号就是你启动外部1编号.

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐