特定GPUにメモリを割り当てる

複数人GPU使用時

特定の実験を特定のGPU上に置くとき、

上記の場合、特定GPUにメモリを割り当てる必要があります.以下のコマンドを使用します.

0番目のGPU(TITAN RTX)を使用したい場合は、

🛠 1.srcコード内で宣言

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

主にtrainコード内で宣言するか、main内でtrain()上で宣言して使用します.

🛠 2.ランタイム宣言

CUDA_VISIBLE_DEVICES=0 python main.py

プログラムの実行時に、割り当てたいGPU番号を宣言することもできます.