VS2017+pyCUDA+CUDA+python安装配置

不念不忘少年蓝@ 2021-11-29 18:24 748阅读 0赞

1、下载vs2017，选择安装C++和python的开发工具  
添加环境变量中 添加path，vs2017 的cl.exe的全路径  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxNzkwODMz_size_16_color_FFFFFF_t_70]  
2、到python官网下载python3或者2，安装。建议使用anaconda进行python 的管理。

3、到CUDA官网下载CUDA，除自动添加的path外，添加新的path  
CUDA\_SDK\_PATH = C:\\ProgramData\\NVIDIA Corporation\\CUDA Samples\\v9.2  
CUDA\_LIB\_PATH = %CUDA\_PATH%\\lib\\x64  
CUDA\_BIN\_PATH = %CUDA\_PATH%\\bin  
CUDA\_SDK\_BIN\_PATH = %CUDA\_SDK\_PATH%\\bin\\win64  
CUDA\_SDK\_LIB\_PATH = %CUDA\_SDK\_PATH%\\common\\lib\\x64  
4、下载PyCuda  
[https://www.lfd.uci.edu/~gohlke/pythonlibs/?cm\_mc\_uid=08085305845514542921829&cm\_mc\_sid\_50200000=1456395916\#pycuda][https_www.lfd.uci.edu_gohlke_pythonlibs_cm_mc_uid_08085305845514542921829_cm_mc_sid_50200000_1456395916_pycuda]  
选择对应版本,cp代表python版本，![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxNzkwODMz_size_16_color_FFFFFF_t_70 1]然后安装

pip install pycuda-2015.1.3+cuda7518-cp35-none-win_amd64.whl

5、vs2017测试  
代码：

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 )

输出：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxNzkwODMz_size_16_color_FFFFFF_t_70 2]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxNzkwODMz_size_16_color_FFFFFF_t_70]: /images/20211129/2f1f60b06c074948aa259675c1161d2d.png
[https_www.lfd.uci.edu_gohlke_pythonlibs_cm_mc_uid_08085305845514542921829_cm_mc_sid_50200000_1456395916_pycuda]: https://www.lfd.uci.edu/~gohlke/pythonlibs/?cm_mc_uid=08085305845514542921829&cm_mc_sid_50200000=1456395916#pycuda
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxNzkwODMz_size_16_color_FFFFFF_t_70 1]: /images/20211129/bd3e281b7f3c4597ba97f3c208c4ab71.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxNzkwODMz_size_16_color_FFFFFF_t_70 2]: /images/20211129/38757ffa0a4b420293834e204f676716.png