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

 

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