YapengMeng / SpecGAN

The official PyTorch implementation of SpecGAN, Large-factor Super-resolution of Remote Sensing Images with Spectra-guided Generative Adversarial Networks (Accept by TGRS).
19 stars 0 forks source link

How to modify the selected GPU number #3

Open likakakaka opened 10 months ago

likakakaka commented 10 months ago

How to modify the selected GPU number?Even though I set the os.environ["CUDA_VISIBLE_DEVICES"], it doesn't work. The GPU number used by the network is still the default 0.