pytorch - Pytorch 用户对旧 GPU 的警告

Question

您好，在上图中，您可以看到我收到有关使用旧 GPU 的警告。我使用以下内容安装了 pytorch 1.6： pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html 现在我运行了一些代码来查看我在计算机上拥有的设备，我发现：

>>> import torch
>>> device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
>>> torch.cuda.get_device_name(0)
'GeForce RTX 2080 Ti'

当我运行这条线时：self.amortized_model.to(device)那是我收到上述警告的时候。现在，根据我运行的 tqdm 条，我每秒的迭代次数也很少。这些东西可能相关吗？有什么方法可以说明训练缓慢的原因是因为它以某种方式使用了错误的 gpu，即使我已经确定设备 0 是我想要的设备？

标签： pytorch