在Jetson TX2和RTX2080TI上面安装pycuda,以及遇到的一些问题解决办法。

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_后边的数字适合你的电脑

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章