Pycuda安装,与pycuda.driver.CompileError的解决

一、pycuda的安装

1、首先确定自己电脑的cuda、cudnn、python版本

2、查看自己应该安装哪个版本，登录网站Pycuda版本网站

前面pycuda-2020.1是版本号，cuda102是cuda版本为10.2，cp36表示python版本为python3.6。
于是自己要安装的版本号。

3、安装之前先导入cuda的路径：

export PATH=/usr/local/cuda/bin:/usr/local/cuda/bin:$PATH

安装命令

pip install pycuda==2020.1

安装完成

二、排坑：自己遇到的错误：此时可能原因是与你的gpu支持的算力不匹配

1.pycuda.driver.CompileError；  nvcc fatal:

解决办法：
修改源码:path:/home/zxb/anaconda3/envs/vortx/lib/python3.9/site-packages/pycuda/compiler.py
#arch = "sm_%d%d" % Context.get_device().compute_capability()
arch = 'sm_70'
将上边的屏蔽，改成下边的直接赋值，前提是sm_后边的数字适合你的电脑

2.[PyCUDA] cuModuleLoadDataEx failed: device kernel image is invalid

解决办法：您正在规定 GPU 架构（arch='...'）。 如果这不匹配您的 GPU，这很容易导致此问题。 只是删除那个kwarg应该没问题。或者改成一个适合自己的arch='sm_XX'。