You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

非Conda环境下使用Numba CUDA时遭遇libNVVM缺失错误的解决求助

非Conda环境下使用Numba CUDA时遭遇libNVVM缺失错误的解决求助

我没有使用Anaconda或Miniconda这类包管理工具,已经从NVIDIA官方渠道下载安装了cuda_12.8.1_572.61_windows.exe,同时手动安装了以下Python库:

  • numba 0.61.0
  • numba-cuda 0.8.0
  • llvmlite 0.44.0
  • numpy 2.1.3
  • cuda-python 12.8.0
  • cuda-bindings 12.8.0
  • pwin32 308

目前我在做程序化生成图片的工作,发现arctan2这类三角函数的计算速度是很大的性能瓶颈。好在我有一台带NVIDIA Geforce GTX 1050 Ti(4GiB显存,后续打算升级)的GPU,所以想通过Numba CUDA把计算转移到GPU上,以此来加速图片生成。

我跟着Numba CUDA的内核开发指南来写代码,但始终跑不起来,每次都会弹出这个错误:

NvvmSupportError: libNVVM cannot be found. Do `conda install cudatoolkit`:
Could not find module 'nvvm.dll' (or one of its dependencies). Try using the full path with constructor syntax.

因为我完全不用Conda,所以没法按照提示安装cudatoolkit。后来我自己找到了对应的dll文件,用PowerShell查询到的路径如下:

PS C:\Users\xenig> get-childitem -path "C:\Program Files\NVIDIA GPU Computing Toolkit" -recurse -filter 'nvvm*.dll'

    Directory: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.8\nvvm\bin

Mode                 LastWriteTime         Length Name
----                 -------------         ------ ----
-a---          2025-02-22    13:18       52873216 nvvm64_40_0.dll

我尝试通过设置环境变量来指定这个dll的路径,但还是没解决问题,代码如下:

import os

os.environ['NUMBAPRO_NVVM'] = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.8/nvvm/bin/nvvm64_40_0.dll"
os.environ['NUMBAPRO_LIBDEVICE'] = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.8/nvvm/libdevice"

现在我实在不知道该怎么处理了,有没有不用Conda的解决办法?


备注:内容来源于stack exchange,提问作者Ξένη Γήινος

火山引擎 最新活动