【发布时间】:2013-09-26 17:05:51
【问题描述】:
仅供参考,我有 64 位版本的 Python 2.7 和 我跟着the pycuda installation instruction安装了pycuda。
而且我运行以下脚本没有任何问题。
import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy
a = numpy.random.randn(4,4)
a = a.astype(numpy.float32)
a_gpu = cuda.mem_alloc(a.nbytes)
cuda.memcpy_htod(a_gpu,a)
但是之后,在执行这条语句的时候,
mod = SourceModule("""
__global__ void doublify(float *a)
{
int idx = threadIdx.x + threadIdx.y * 4;
a[idx] *= 2;
}
""")
我收到错误消息
CompileError: nvcc 编译 c:\users\xxxx\appdata\local\temp\tmpaoxt97\kernel.cu 失败 [命令:nvcc --cubin -arch sm_21 -m64 -Ic:\python27\lib\site-packages\pycuda\cuda kernel.cu] [stderr: nvcc: 致命错误: nvcc 找不到支持的 Microsoft Visual 版本 工作室。仅支持 2008、2010 和 2012 版本
但我在机器上安装了 VS 2008 和 VS 2010,并按照说明设置路径和 nvcc 配置文件。有人告诉我这是怎么回事吗?
UPDATE1:正如 cgohike 指出的,在有问题的语句之前运行以下语句将解决问题。
import os
os.system("vcvarsamd64.bat")
【问题讨论】:
-
您是否安装了 Visual Studio 2008 或更高版本?