How do I fix this?AttributeError: 'module' object has no attribute '_cuda_setDevice'

[XinChengit]

I am running this on a MacBook Pro (Retina, 13-inch, Early 2015) macOS Catalina 10.15.3 GPU Intel Iris Graphics 6100 1536 MB
I am at this step in the instructions: python main.py --cfg cfg/birds_3stages.yml --gpu 0

And I get this as a result:
/Users/f00l/Desktop/GitHubProject/StackGAN-v2/code/miscc/config.py:103: YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Please read https://msg.pyyaml.org/load for full details.
yaml_cfg = edict(yaml.load(f))
Using config:
{'CONFIG_NAME': '3stages',
'CUDA': False,
'DATASET_NAME': 'birds',
'DATA_DIR': '../data/birds',
'EMBEDDING_TYPE': 'cnn-rnn',
'GAN': {'B_CONDITION': True,
'DF_DIM': 64,
'EMBEDDING_DIM': 128,
'GF_DIM': 64,
'NETWORK_TYPE': 'default',
'R_NUM': 2,
'Z_DIM': 100},
'GPU_ID': '0',
'TEST': {'B_EXAMPLE': True, 'SAMPLE_NUM': 30000},
'TEXT': {'DIMENSION': 1024},
'TRAIN': {'BATCH_SIZE': 24,
'COEFF': {'COLOR_LOSS': 0.0, 'KL': 2.0, 'UNCOND_LOSS': 1.0},
'DISCRIMINATOR_LR': 0.0002,
'FLAG': True,
'GENERATOR_LR': 0.0002,
'MAX_EPOCH': 600,
'NET_D': '',
'NET_G': '',
'SNAPSHOT_INTERVAL': 2000,
'VIS_COUNT': 64},
'TREE': {'BASE_SIZE': 64, 'BRANCH_NUM': 3},
'WORKERS': 4}
/Users/f00l/Desktop/GitHubProject/StackGAN-v2/venv/lib/python2.7/site-packages/torchvision/transforms/transforms.py:156: UserWarning: The use of the transforms.Scale transform is deprecated, please use transforms.Resize instead.
"please use transforms.Resize instead.")
Total filenames: 11788 001.Black_footed_Albatross/Black_Footed_Albatross_0046_18.jpg
Load filenames from: ../data/birds/train/filenames.pickle (8855)
embeddings: (8855, 10, 1024)
Traceback (most recent call last):
File "main.py", line 140, in
algo = trainer(output_dir, dataloader, imsize)
File "/Users/f00l/Desktop/GitHubProject/StackGAN-v2/code/trainer.py", line 537, in init
torch.cuda.set_device(self.gpus[0])
File "/Users/f00l/Desktop/GitHubProject/StackGAN-v2/venv/lib/python2.7/site-packages/torch/cuda/init.py", line 262, in set_device
torch._C._cuda_setDevice(device)
AttributeError: 'module' object has no attribute '_cuda_setDevice'

How do I fix this?

[mansijoshii]

Before torch.cuda.setDevice in the trainer.py code, try adding an if condition. Do it only if cfg.CUDA is true.