深度学习CUDA报错合集:

9 篇文章 0 订阅
2 篇文章 0 订阅

1.RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

原因:当前电脑GPU算力与当前环境下pytorch中的cuda算力不匹配

解决:当前shell下执行:

export TORCH_CUDA_ARCH_LIST="8.0"        # 数值匹配于cuda版本

ps:torch常用命令:

python
import torch
torch.__version__
torch.cuda.is_availabel()
torch.cuda.get_device_capability()

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
当你在使用Jupyter Notebook时遇到"AssertionError: Torch not com with CUDA enabled"的错误,这通常是因为你的PyTorch库没有启用CUDA支持。 CUDA是NVIDIA提供的并行计算平台,它可以利用GPU的强大计算能力来加速深度学习任务。PyTorch是一个基于Torch的深度学习框架,它也支持CUDA加速。 要解决这个问题,你可以按照以下步骤进行操作: 1. 确保你的计算机上已经正确安装了NVIDIA显卡驱动和CUDA工具包。你可以在NVIDIA官方网站上下载并安装最新版本的驱动程序和CUDA工具包。 2. 确保你已经正确安装了PyTorch库。你可以使用以下命令来安装PyTorch: ``` pip install torch ``` 如果你想安装支持CUDA的PyTorch版本,可以使用以下命令: ``` pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html ``` 这将安装PyTorch 1.9.0版本,并且支持CUDA 11.1。 3. 在Jupyter Notebook中,确保你选择了正确的内核。你可以在Jupyter Notebook的菜单栏中选择"Kernel" -> "Change Kernel",然后选择一个与你安装的PyTorch版本相对应的内核。 如果你按照以上步骤操作后仍然遇到相同的错误,可能是因为你的计算机上没有可用的CUDA设备。在这种情况下,你可以尝试在没有CUDA支持的情况下使用CPU进行计算,或者考虑在具有CUDA支持的计算机上运行代码。 希望以上信息对你有帮助!如果你还有其他问题,请随时提问。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值