成功解决:RuntimeError: CUDA error: no kernel image is available for execution on the device
成功解决:RuntimeError: CUDA error: no kernel image is available for execution on the device
在跑DDPM项目时,出现如下输出:
RuntimeError: CUDA error: no kernel image is available for execution on the device
原因是: CUDA 版本与 GPU 驱动程序或 PyTorch 安装的 CUDA 编译版本不兼容
如何解决:
安装的 PyTorch 版本和 CUDA 版本不兼容。可以重新安装与系统上 CUDA 版本匹配的 PyTorch 版本。以下是一些常见的安装命令:
CUDA版本查看命令:
nvcc -V
输出:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0
需要注意命令中的118换成你的CUDA版本
使用 pip 安装 PyTorch:
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple
使用 conda 安装 PyTorch:
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch
验证pytorch版本和CUDA版本:
import torch
print(torch.__version__)
print(torch.version.cuda)
输出:

安装验证完成后,就可以完美解决这个Bug!!!
更多推荐
所有评论(0)