fix variable name and import errors; use cpu when no nvidia gpu.

[userElaina]

fix: variable name and import errors.

If num_workers=2 in torch.utils.data.DataLoader(), an error will occur without nvidia gpu. So I let num_workers=2 if torch.cuda.is_available() else 0 to run use cpu when no nvidia gpu.