当前位置:   article > 正文

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

raise assertionerror("torch not compiled with cuda enabled")

下图确认是否有cuda

  1. import torch
  2. print(torch.__version__)
  3. print(torch.cuda.is_available())

若为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"

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/707989
推荐阅读
相关标签
  

闽ICP备14008679号