一、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:
解决办法: