错误再现：
出错原因：
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
暂无评论