pandazx's blog

データ分析など雑多な技術ブログ

Dockerコンテナで利用可能なGPU制御

マルチGPUサーバで、あるコンテナには指定GPUのみ利用を許可したい場合。

環境変数 NVIDIA_VISIBLE_DEVICESでGPUバイスIDを指定して、コンテナを起動すればよい。

docker run -d -it --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES='1' --name {image_name} /bin/bash

参考 - http://cut-ter.hatenablog.com/entry/2017/12/27/161547