module 'torch' has no attribute 'device'
时间: 2024-04-24 17:18:56 浏览: 198
导入FashionMNIST数据集时报错module ‘torchvision.datasets’ has no attribute ‘FashionMNIS’
根据提供的引用内容,你遇到的问题是关于`module 'torch' has no attribute 'device'`的错误。根据引用和引用的解决方法,你可以尝试以下两种方法来解决这个问题:
1. 将`with torch.autocast(device.type, enabled=use_fp16):`改为`with torch.cuda.amp.autocast(enabled=use_fp16):`[^1]。
2. 确保你的PyTorch版本在1.6.0以上,并且你的CUDA版本是11.6,同时你的Python版本是3.8。如果你的CUDA版本过旧,可能会导致出现`cuda driver too old`的错误信息。你可以尝试升级你的CUDA驱动程序来解决这个问题。
阅读全文