报错AssertionError: Torch not compiled with CUDA enabled及解决办法,cuda安装失败个人解决办法,jupyter内核挂掉解决方法

下图确认是否有cuda

若为false,说明我们需要安装cuda

首先需要在命令行输入nvidia-smi,查看个人电脑可使用的cuda版本

我在命令提示符中直接下载的torch为2.2版本,根据适配的版本,我最开始选择的是cuda12.1版本,但在最后安装的时候出现安装失败的情况(此处忘记截图了)

(此时注意可以在电脑控制面板-程序-程序与功能-在搜索栏输入NVIDIA卸载掉已下载的部分,再下载其他版本)

经过网上的查阅,最终选择下载cuda11.8版本,这个版本可以正常安装,再通过pytorch官网找到对应的pytorch版本的命令,

下图是pytorch下载官网中的选择

-----------------------------------------------------------------------------------------------------------------------------(补充)安装成功后,jupyter查看torch版本仍为2.2.0,即使是在命令行中卸载pytorch也没有用,此时直接在文件夹搜索栏,搜索pytorch,找到2.2.0版本文件,删除即可

------------------------------------------------------------------------------------------------------------------------------

重启jupyter,又出现jupyter挂掉的情况,网上说输入

import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"

可以解决此问题,但问题依旧存在,最后是重启大法成功解决(个人认为是清除了加载缓存,解决重复加载的问题,因此可以恢复运行)

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值