这个原因是pytorch不支持CUDA,
可以先输入
import torch
print(torch.cuda.is_available())
如果输入为false则打开cmd,输入nvidia-smi查看cuda的版本,之后去Previous PyTorch Versions | PyTorch找符合自己cuda的下载方式,重新下载一遍即可
这个原因是pytorch不支持CUDA,
可以先输入
import torch
print(torch.cuda.is_available())
如果输入为false则打开cmd,输入nvidia-smi查看cuda的版本,之后去Previous PyTorch Versions | PyTorch找符合自己cuda的下载方式,重新下载一遍即可
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。