search

MIVRC / SeaNet-PyTorch

This repository is a PyTorch version of "Soft-edge Assisted Network for Single Image Super-Resolution". (IEEE TIP 2020)
27 stars 10 forks source link

怎么控制GPU的序号 #3

Open zkh123456789 opened 3 years ago

zkh123456789 commented 3 years ago

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

CV-JunchengLi commented 2 years ago

在trainer.py加

import os

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