SeaNet-PyTorch icon indicating copy to clipboard operation
SeaNet-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon MIVRC

怎么控制GPU的序号

Open zkh123456789 opened this issue 3 years ago • 1 comments

作者您好，我想知道训练的时候怎么指定GPU的型号，每次我使用指令CUDA_VISIBLE_DEVICES=2,3这条指令的时候，似乎不管用，还是使用的默认的前两块GPU，请问您如何解决这个问题。

zkh123456789 avatar Sep 26 '21 11:09 zkh123456789

在trainer.py加

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "4,5"

CV-JunchengLi avatar Oct 27 '21 07:10 CV-JunchengLi