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_後邊的數字適合你的電腦