docker18.09.3版本创建容器时指定gpu号
总会遇到各种想不到的情况
在a处测试服务可用,拿到b处发现无法运行,分析了一会,才知道gpu显存满了,因为默认使用0号gpu,所以需要重新指派其他gpu,于是,按照正常docker命令加上--gpus就可以了
可偏偏b处的docker是18版本,不支持这个参数,擦,还不能升级
查了一圈又一圈,不顶用的文档多,最后在创建时传递一个环境变量进去进行,怎么一开始没想到,于是
docker run -e NVIDIA_VISIBLE_DEVICES=1 -it your_image_name
这样就可以使用1号gpu了,真是应了那句话,属于的理论我都知道,但是遇到问题还是想不到答案,呵呵
本文来自博客园,作者:海_纳百川,转载请注明原文链接:https://www.cnblogs.com/chentiao/p/17986665,如有侵权联系删除

浙公网安备 33010602011771号