python安装cuda出现错误提示:AssertionError Torch not compiled with CUDA enabled(1)

如果你的电脑中有CUDA但版本不合适,删除重装即可,需要注意的是,一定要卸载干净。我是在电脑自带的控制面板中进行卸载的:点击控制面板——程序——程序和功能。找到所有和NVIDIA有关的全部卸载,然后再去软件管家清理卸载残留。其他的卸载方式也可,但一定要卸载干净。

在这里插入图片描述

安装完成后,检查是否安装成功,再次打开cmd,输入

nvcc -V

2.pytorch环境的配置

首先对照CUDA和pyTorch对应的版本:
在这里插入图片描述

选择好版本后我们就开始下载pytorch安装包
方法有两种:
1.在pytorch官网:https://pytorch.org/get-started/locally/选择在线安装,接着访问 Pytorch 的官网,找到与你本机 CUDA 对应版本的 Pytorch,并进行安装,具体操作如下:
在这里插入图片描述
若你的 CUDA 版本没有出现在官网给出可选的选项范围内,则进入下载其他版本的入口进行历史版本的下载:
在这里插入图片描述

2.离线安装
参考文章:https://blog.csdn.net/weixin_72056722/article/details/126651703?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-126651703-blog-125737765.pc_relevant_multi_platform_whitelistv3&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-126651703-blog-125737765.pc_relevant_multi_platform_whitelistv3&utm_relevant_index=1

以上安装完成后,可以进行简单的版本检查测试:

>>> improt torch

>>> print(torch.__version__)

>>> print(torch.cuda.is_available())

最后输出的结果如果是:

1.11.0+cu113

True

恭喜你,安装成功!
但如果是像我这样的倒霉蛋😀:可能是这样的:
在这里插入图片描述
建议直接检查显卡型号:
在设备管理器中,win+x选择“设备管理器”,找到“显示适配器”,
在这里插入图片描述
如果出现上面的样子,建议换个机子或者外接显卡或外接服务器了(* ^ _ ^ *)
如果有NVIDIA的显卡话,可以尝试重新安装一下显卡驱动:
参考文章:https://blog.csdn.net/qq_45507678/article/details/124130123
NVIDIA官网:https://www.nvidia.cn/geforce/drivers/
在这里插入图片描述
第二个选项栏目是所有的型号系列
例如1050显卡
台式电脑选择GeForce 10 Series,笔记本电脑选择GeForce 10 Series(Notebooks)

还有一种倒霉蛋,电脑是Macbook的,就别想了,苹果好像和英伟达闹掰了,不支持,外接吧,或者服务器,别问我为什么知道(* ^ _ ^ *)

cuda的卸载重装:

网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友,可以戳这里无偿获取

一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!

  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值