Jetson TX2上面安装pycuda:
sudo pip3 install pycuda
安装过程中不会报错但是在测试demo的时候会报无法找到nvcc路径问题。找到compiler.py文件,直接将ncvv的路径改为绝对路径就可以了。
RTX2080TI安装pycuda:使用的是anaconda:
pip install pycuda
安装过程中也不会报错,运行demo就会出现问题。nvcc fatal :Value 'sm_75'is not defined for option 'gpu_architechture'
解决办法也是找到compiler.py文件,将里面的替换掉就可以如下所示:
#arch = "sm_%d%d" % Context.get_device().compute_capability()
arch = 'sm_70'
将上边的屏蔽,改成下边的直接赋值,前提是sm_后边的数字适合你的电脑