在跑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!!!

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐