解决报错：RuntimeError: Invalid device string: ‘cuda：0‘

错误再现：

出错原因：

os.environ['CUDA_VISIBLE_DEVICES'] = '0'
device = torch.device('cuda：0' if torch.cuda.is_available() and use_cuda else 'cpu')

应修改为：

os.environ['CUDA_VISIBLE_DEVICES'] = '0'
device = torch.device('cuda' if torch.cuda.is_available() and use_cuda else 'cpu')

至于为什么这样改，还不清楚，欢迎大佬评论区留言！

版权声明：本文为CSDN博主「零碎@流年絮语」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_44824148/article/details/121021722