安装方式:
在终端输入
sudo apt-get install python3-pycuda
安装完成 运行 程序可能会发生以下错误:
lab30202@lab30202:~/xg$ python3 test.py
Traceback (most recent call last):
File "test.py", line 11, in <module>
import pycuda.autoinit
File "/usr/lib/python3/dist-packages/pycuda/autoinit.py", line 5, in <module>
cuda.init()
pycuda._driver.Error: cuInit failed: unknown error
这是cuda初始化的一些原因,
在输入sudo apt-get install nvidia-modprobe命令
安装完就可以用了