pycuda的安装和错误调试

最新推荐文章于 2025-06-18 09:00:04 发布

原创最新推荐文章于 2025-06-18 09:00:04 发布 · 3.6k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#pycuda

linux环境配置专栏收录该内容

3 篇文章

订阅专栏

本文详细介绍了在特定的计算机配置下（CUDA 9.0, cuDNN 7.4.2, Python 3.5）安装PyCUDA过程中遇到的错误及解决方案。包括修改compiler.py文件以解决'nvcc'未找到的问题，以及如何调整arch参数以解决编译错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

计算机各配置版本信息:
cuda:9.0
cudnn:7.4.2
python:3.5

pycuda安装:
pip install pycuda==2017.1.1

安装后运行报错:FileNotFoundError: [Errno 2] No such file or directory: ‘nvcc’
修改compiler.py文件中的compile_plain()函数如下得以解决，如下:
def compile_plain(source, options, keep, nvcc, cache_dir, target=“cubin”):
　　from os.path import join
　　assert target in [“cubin”, “ptx”, “fatbin”]
　　nvcc = ‘/usr/local/cuda/bin/’+nvcc
　　if cache_dir:
　　　　checksum = new_md5()
参考链接:　https://codeday.me/bug/20190321/796820.html
之后又出现另一个问题:
pycuda.driver.CompileError: nvcc compilation of /tmp/tmpr0d7p_uv/kernel.cu failed
我通过参考博客:　https://blog.youkuaiyun.com/weixin_38730719/article/details/90320632
将compiler.py文件中的arch = "sm%d%d"改成arch = 'sm_70’即可
同时推荐一本pycuda的学习教材:“GPU-Accelerated Computer Vision with Opencv and CUDA”