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 <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命令

安装完就可以用了



阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页