Semihal
9/26/2019 - 10:45 AM

Set used GPU

utils.py
content_copyfile_download
import os

def set_gpu(num):
    gpu = str(num)
    os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
    os.environ["CUDA_VISIBLE_DEVICES"] = gpu