ubuntu 下pycuda的安装

安装方式:

在终端输入
sudo apt-get install python3-pycuda

安装完成 运行 程序可能会发生以下错误:

lab30202@lab30202:~/xg$ python3 test.py
Traceback (most recent call last):
File "test.py", line 11, in
import pycuda.autoinit
File "/usr/lib/python3/dist-packages/pycuda/autoinit.py", line 5, in
cuda.init()
pycuda._driver.Error: cuInit failed: unknown error

这是cuda初始化的一些原因,

在输入sudo apt-get install nvidia-modprobe命令

安装完就可以用了



你可能感兴趣的:(pycuda)