Python GPU3:Pycuda

0. 准备

1. 安装

1.1 方法一

pip install pycuda

1.2 方法二

1. 访问网址，下载

Python Extension Packages for Windows - Christoph Gohlke

The binaries are compatible with the most recent official CPython distributions on Windows >=6.0. Chances are they do not work with custom Python distributions included with Blender, Maya, ArcGIS, OSGeo4W, ABAQUS, Cygwin, Pythonxy, Canopy, EPD, Anaconda, WinPython etc. Many binaries are not compatible with Windows XP, Windows 7, or Wine.

https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

1. 进入网站

1. 下拉，找到适合自己Python解释器和CUDA的pycuda版本，我机器信息
• CUDA11.4
• python3.8
• 64位

1. 文件信息解释

• pycuda‑2021.1+cuda114‑cp38‑cp38m‑win_amd64.whl
• pycuda‑2021.1：表示pycuda版本是2021.1
• cuda114：表示CUDA版本为11.4
• cp38‑cp38m：Python解释器3.8
• win_amd64.whl：64位

1. 安装

# 安装
pip install pycuda-2021.1+cuda114-cp38-cp38-win_amd64.whl
# 卸载
pip uninstall pycuda-2021.1+cuda114-cp38-cp38-win_amd64.whl

pip install -U typing-extensions==3.7.4

pip install -U pytools==2021.2.7

2. 测试

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy

# 创建了一个随机数组成的4*4大小的数组a
a = numpy.random.randn(4, 4)
# 创建的这个数组a包含的是双精度浮点数，但大多数常用的NVIDIA显卡只支持单精度浮点数，所以需要转换一下类型
a = a.astype(numpy.float32)
# 已有的数据转移过去，还要设定一个目的地，所以我们要在显卡中分配一段显存
a_gpu = cuda.mem_alloc(a.nbytes)
# 生成的数组a转移到GPU
cuda.memcpy_htod(a_gpu, a)

# 编译代码，并且加载到显卡中，把显存中的数组a_gpu作为参数传过去，同时设定块大小为4x4：
mod = SourceModule("""
  __global__ void doublify(float *a)
  {
    int idx = threadIdx.x + threadIdx.y*4;
    a[idx] *= 2;
  }
  """)
# 使用pycuda.driver.Function,然后调用此引用
func = mod.get_function("doublify")
# 显存中的数组a_gpu作为参数传过去，同时设定块大小为4x4
func(a_gpu, block=(4, 4, 1))
# 把经过运算处理过的数据从GPU取回，并且将它和原始数组a一同显示出来对比
a_doubled = numpy.empty_like(a)
cuda.memcpy_dtoh(a_doubled, a_gpu)
print(a_doubled)
print(a)

pip install -U numpy

2.1 下载Visual Studio 2022

C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\14.31.31103\bin\Hostx64\x64

2.2 下载Visual Studio 2019

Thank you for downloading Visual Studio - Visual Studio

To develop any type of app or learn a language, you'll be working in the Visual Studio Integrated Development Environment (IDE). Beyond code editing, Visual Studio IDE brings together graphical designers, compilers, code completion tools, source control, extensions and many more features in one place.

https://visualstudio.microsoft.com/thank-you-downloading-visual-studio/?sku=Community&rel=16

C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.29.30133\bin\Hostx64\x64

环境变量设置

参考

pycuda报错 nvcc fatal : Cannot find compiler 'cl.exe' in PATH_我是狮子搏兔的博客-CSDN博客

报错信息如下：原因是没找到C++的编译器，C++编译器安装请看上编博客安装之后，需要更改一下环境变量，桌面右键我的电脑，选择属性

https://blog.csdn.net/qq_40811682/article/details/118033631

Pycharm安装pycuda出现错误 Microsoft Visual C++ 14.0 is required. Get it with "Microsoft Visual

装之前首先电脑得有GPU吧，如果只要核显可能也装不上，但不是这个错误，当然就算装上了没有GPU也没用。 我在个人笔记本上，显卡为RTX2060，使用Pycharm安装pycuda包，出现 Microsoft Visual C++ 14.0 is required. Get it with "Microsoft Visual C++ Build Tools 去这个网址下一下最新的C++就行了，大概2G，官方网址放心点。 https://visualstudio.microsoft.com/zh-hans/downloads/ 选择单个组件，只安装下面这个就行了！ 下载之后需要 配置下环境变量！桌面右键我的电脑，点击属性。 使用默认安装，参考环境变量为： C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Tools\MSVC\14.16.27023\bin\Hostx64\x64 具体如果安装自定义以你自己的安装路径为准

https://blog.csdn.net/qq_40811682/article/details/118033177?spm=1001.2014.3001.5501