Table of Contents

DelurGan Training FAQ

Q: `AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'`

A: This is caused by running on a CPU-only platform while the training command is trying to select a CUDA device.

Add --gpu_ids -1 to the command so the project runs in CPU mode:

python train.py --gpu_ids -1

If you expected to train on GPU, verify that your local PyTorch build includes CUDA support before changing the command:

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)

A: Import ssim from the ssim package and alias it as SSIM:

from ssim import ssim as SSIM

This fixes code that expects the name SSIM while the installed module exposes the function as ssim.