本文介绍了遇到PyTorch GPU运行错误时的解决方法，重点检查CUDA版本与torch版本的一致性。确保torch的cudatoolkitversion与CUDA Version匹配，从PyTorch官网下载cu10.0开头的对应版本，并通过命令行进行安装。如果已安装，可以尝试强制重装。此方案可解决本质问题，而非表面现象。
import torch
torch.version.cuda查看具体的cuda版本号，检测其版本号是否跟显卡的cuda是否一致
从图中可以看出，CUDA Version为10.0
因此torch的cudatoolkitversion也必须为10.0（最好保持一致，但不能高于，我这儿就是版本低于显卡的CUDA版本而报错的）
然后去官网下载对应torch版本（使用whl，使用pip或者conda命令安装会出问题）：
https://download.pytorch.org/whl/torch_stable.html
找打是cu10.0开头的版本torch 别的会有问题，然后下载到本地进行命令安装
2万+
2892
1173
1423
2026
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
最低0.47元/天 解锁文章
扫一扫
