import pycuda.driver as drv
import pycuda.tools
import pycuda.autoinit
import numpy
import numpy.linalg as la
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))

print dest-a*b

C:\Python27>hello_gpu.py
Traceback (most recent call last):
File “C:\Python27\hello_gpu.py”, line 14, in <module>
“”“)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py“, line 235, in __init__
arch, code, cache_dir, include_dirs)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py“, line 201, in compile
”pycuda-compiler-cache-v1-%s“ % _get_per_user_string())
File ”C:\Python27\lib\site-packages\pycuda\compiler.py", line 139, in _get_per_user_string
checksum.update(environ)
File “C:\Python27\lib\os.py”, line 423, in __getitem__
return self.data
KeyError: ‘HOME’

def _get_per_user_string():
    try:
        from os import getuid
    except ImportError:
        checksum = _new_md5()
        from os import environ
        checksum.update(environ["HOME"])
        return checksum.hexdigest()
    else:
        return "uid%d" % getuid()

guranvir
обратите внимание, что возможно вам нужен PYTHONPATH

guranvir
Просто насколько я понимаю в PYTHONHOME хранится путь до интерпретатора. Но обычно для этих целей устанавливают переменую PYTHONPATH. НО возможно я ошибаюсь, то же это учтите))

C:\Python27>hello_gpu.py
Traceback (most recent call last):
File “C:\Python27\hello_gpu.py”, line 14, in <module>
“”“)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py“, line 236, in __init__
arch, code, cache_dir, include_dirs)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py“, line 226, in compile
return compile_plain(source, options, keep, nvcc, cache_dir)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py", line 107, in compile_plain
cmdline, stdout=stdout, stderr=stderr)
pycuda.driver.CompileError: nvcc compilation of c:\users\aace~1\appdata\local\temp\tmp_dyt87\kernel.cu failed

[stderr:
nvcc fatal : Cannot find compiler ‘cl.exe’ in PATH
]

guranvir
cl.exe входит в состав MS Visual Studio и является компилятором управляемым через командную строку, бесплатно доступна Visual Studio Express. А так вобще этот вопрос гуглится неплохо))

Traceback (most recent call last):
File “C:\Python27\hello_gpu.py”, line 14, in <module>
“”“)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py“, line 235, in __init__
arch, code, cache_dir, include_dirs)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py“, line 225, in compile
return compile_plain(source, options, keep, nvcc, cache_dir)
File ”C:\Python27\lib\site-packages\pycuda\compiler.py", line 107, in compile_plain
cmdline, stdout=stdout, stderr=stderr)
pycuda.driver.CompileError: nvcc compilation of c:\users\3d47~1\appdata\local\temp\tmpnmw0vl\kernel.cu failed

[stderr:
nvcc fatal : nvcc cannot find a supported cl version. Only MSVC 8.0 and MSVC 9.0 are supported
]