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了,真是应了那句话,属于的理论我都知道,但是遇到问题还是想不到答案,呵呵

posted @ 2024-01-25 10:46  海_纳百川  阅读(396)  评论(0)    收藏  举报
本站总访问量