torch.cuda.is_available()返回false?

问题描述:

>>python
>>import torch
>>torch.cuda.is_available()
>>false

解决措施:

1.检查自己的GPU是否支持cuda.

这一步可以上英伟达的官网上,输入自己的GPU型号即可查看。

2.更新驱动。

可以通过软件管家升级或者去英伟达的官网上下载最新的驱动。

3.可以通过在控制台输入 nvidia -smi 获取自己的驱动信息。

你可能感兴趣的:(#,pytorch,深度学习,pytorch,transformer)