Pycuda -如何添加-ccbin clang-3.8
Pycuda -如何添加-ccbin clang-3.8
提问于 2018-03-23 12:09:02
我目前正试图在Debian 9上使用PyCUDA。我已经设法使cuda工作了，如果我运行：
nvcc -ccbin clang-3.8 file.cu我正确地编译了这个文件，并且能够运行它。
但是，在我使用pycuda之后
apt-get install python-pycuda然后在他们的网站上运行一个简单的例子：
import pycuda.autoinit
import pycuda.driver as drv
import numpy
from pycuda.compiler import SourceModule
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
const int i = threadIdx.x;
dest[i] = a[i] * b[i];
}
""")
multiply_them = mod.get_function("multiply_them")
a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)
dest = numpy.zeros_like(a)
multiply_them(
drv.Out(dest), drv.In(a), drv.In(b),
block=(400,1,1), grid=(1,1))
print dest-a*b但我收到以下错误：
CompileError Traceback (most recent call last)
<ipython-input-1-8e16128de7f2> in <module>()
10 dest[i] = a[i] * b[i];
11 }
---> 12 """)
13
14 multiply_them = mod.get_function("multiply_them")
/usr/lib/python2.7/dist-packages/pycuda/compiler.pyc in __init__(self, source, nvcc, options, keep, no_extern_c, arch, code, cache_dir, include_dirs)
263
264 cubin = compile(source, nvcc, options, keep, no_extern_c,
--> 265 arch, code, cache_dir, include_dirs)
266
267 from pycuda.driver import module_from_buffer
/usr/lib/python2.7/dist-packages/pycuda/compiler.pyc in compile(source, nvcc, options, keep, no_extern_c, arch, code, cache_dir, include_dirs, target)
253 options.append("-I"+i)
254
--> 255 return compile_plain(source, options, keep, nvcc, cache_dir, target)
256
257
/usr/lib/python2.7/dist-packages/pycuda/compiler.pyc in compile_plain(source, options, keep, nvcc, cache_dir, target)
135 raise CompileError("nvcc compilation of %s failed" % cu_file_path,
136 cmdline, stdout=stdout.decode("utf-8", "replace"),
--> 137 stderr=stderr.decode("utf-8", "replace"))
138
139 if stdout or stderr:
CompileError: nvcc compilation of /tmp/tmpVgfyrm/kernel.cu failed
[command: nvcc --cubin -arch sm_61 -I/usr/local/lib/python2.7/dist-packages/pycuda-2017.1.1-py2.7-linux-x86_64.egg/pycuda/cuda kernel.cu]
[stderr:
ERROR: No supported gcc/g++ host compiler found, but clang-3.8 is available.
Use 'nvcc -ccbin clang-3.8' to use that instead.
]有人知道我怎么能把-ccbin clang-3.8添加到pycuda中吗？？
回答 2
Stack Overflow用户
回答已采纳
发布于 2018-03-23 18:08:17
根据文档，可以通过两种方式为nvcc指定编译器选项。
- 通过
PYCUDA_DEFAULT_NVCC_FLAGS环境变量设置默认编译器选项。 - 通过使用
SourceModule关键字传递的列表为给定的options=设置编译器选项
Stack Overflow用户
发布于 2018-03-23 21:27:10
对于每一个有问题的人，使用选项参数给出的解决方案就是给出的方法。我使用的代码如下：
import pycuda.autoinit
import pycuda.driver as drv
import numpy
from pycuda.compiler import SourceModule
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
const int i = threadIdx.x;
dest[i] = a[i] * b[i];
}
""", options=["-ccbin","clang-3.8"])
multiply_them = mod.get_function("multiply_them")
a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)
dest = numpy.zeros_like(a)
multiply_them(
drv.Out(dest), drv.In(a), drv.In(b),
block=(400,1,1), grid=(1,1))
print dest-a*b或使用：
pycuda.compiler.DEFAULT_NVCC_FLAG = ["-ccbin","clang-3.8"]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/49449355
复制相关文章
相似问题
腾讯云开发者
