电脑睡眠之后程序停止,提示CUDA unknown error

文章描述了在使用PyTorch时遇到CUDAinitialization未知错误的情况,警告可能由于环境设置不当,如更改CUDA_VISIBLE_DEVICES。解决方案是通过`sudomodprobenvidia_uvm`和`sudomodprobepnp_nvidia`来修复。
部署运行你感兴趣的模型镜像

电脑睡眠之后程序停止,提示CUDA unknown error

报错内容

>>> print(torch.cuda.is_available())
/home/lab/anaconda3/envs/lzp_torch/lib/python3.10/site-packages/torch/cuda/__init__.py:124:
UserWarning: CUDA initialization: CUDA unknown error - this may be due
to an incorrectly set up environment, e.g. changing env variable
CUDA_VISIBLE_DEVICES after program start. Setting the available
devices to be zero. (Triggered internally at
/opt/conda/conda-bld/pytorch_1681197222248/work/c10/cuda/CUDAFunctions.cpp:108.)
  return torch._C._cuda_getDeviceCount() > 0
False
>>>

解决办法

sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm

参考:https://discuss.pytorch.org/t/userwarning-cuda-initialization-cuda-unknown-error-this-may-be-due-to-an-incorrectly-set-up-environment-e-g-changing-env-variable-cuda-visible-devices-after-program-start-setting-the-available-devices-to-be-zero/129335/4

您可能感兴趣的与本文相关的镜像

PyTorch 2.7

PyTorch 2.7

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

回答: 当出现CUDA unknown error时,可能是由于环境设置不正确导致的。比如在程序启动后更改了CUDA_VISIBLE_DEVICES环境变量。这可能会导致可用设备设置为零。\[1\]另外,报错信息中还提到了CUDA kernel errors可能会在其他API调用时异步报告,因此下面的堆栈跟踪可能是不正确的。为了调试,可以考虑传递CUDA_LAUNCH_BLOCKING=1参数。\[2\]此外,在WSL2下的Ubuntu 20.04上运行YOLOv8时也可能出现相同的错误。\[3\]要解决这个问题,可以尝试重新设置环境变量,确保正确配置CUDA_VISIBLE_DEVICES,并考虑传递CUDA_LAUNCH_BLOCKING=1参数进行调试。 #### 引用[.reference_title] - *1* [CUDA unknown error - this may be due to an incorrectly set up environment 问题解决](https://blog.csdn.net/zhiweihongyan1/article/details/122241982)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [debug——偶发报警:RuntimeError: CUDA error: unknown error](https://blog.csdn.net/wxsy024680/article/details/131450595)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [【YOLO】CUDA error: unknown error](https://blog.csdn.net/weixin_42166222/article/details/129403517)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值