1. Python

RuntimeError: cuda runtime error (48) : no kernel image is available for execution on the device at …

最近使用VIBE,按照其官网使用conda安装环境,但是在运行的时候出现如下错误:

RuntimeError: cuda runtime error (48) : no kernel image is available for execution on the device at /pytorch/aten/src/THCUNN/generic/LeakyReLU.cu:29

查看其自动安装的环境,发现没有安装带cuda的torch,只好自己来安装,试了一些低版本的torch和cuda的搭配,还是会有上面的错误。最后找了如下版本的:

pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116

问题解决。