cuda-python 安装没有匹配的版本(for cuda11)

这个问题困扰了我两天,今天下定决心分析一下, 因为我有一个作业必须要依赖cuda-python,其实用pycuda也是可以的,但是要修改很多代码, 我对pycuda的api又不是很熟悉,所以决定还是搞定cuda-python

nvidia官方的网站是只有cuda12的安装教程,但是我的环境是cuda11的,但是在cuda11的环境下安装pip install cuda-python 是报了no matching的错误, 所以一直卡在这

今天发现,cuda-python的github实际上是有以前cuda11的release的source code,那么直接从source code 编译这个库就行了。

Release v11.4.0 · NVIDIA/cuda-python · GitHub

编译的步骤看readme很简单, 安装好cython,直接python setup.py develop 

运行一下就编译并且安装成功了

 哈哈, 成功导入 ,搞定

猜你喜欢

转载自blog.csdn.net/qq_31638535/article/details/128690685