cublas library not found

最新推荐文章于 2024-07-09 11:31:41 发布

风尘23187

最新推荐文章于 2024-07-09 11:31:41 发布

阅读量815

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/ygf666/article/details/123174113

版权

在Windows环境下，作者遇到使用skcuda库时由于缺少cublas库导致的错误。通过手动指定cublas64_10.dll文件路径，成功解决了OSError：cublas library not found的问题。修改了skcuda的源代码中对应平台的库名称列表，使得程序能够正确找到所需的dll文件并运行。

摘要由CSDN通过智能技术生成

在使用skcuda，运行以下代码时

import skcuda.linalg as linalg

报错：

File "C:\Users\victo\Anaconda3\envs\spikesorting_tsne\lib\site-packages\skcuda\linalg.py", line 21, in
from . import cublas

File "C:\Users\victo\Anaconda3\envs\spikesorting_tsne\lib\site-packages\skcuda\cublas.py", line 55, in
raise OSError('cublas library not found')

OSError: cublas library not found

我的解决方案：
用everything，搜索cublas，找到cublas**.dll，比如cublas64_10.dll，然后将报错的cublas.py中的以下部分：

if 'linux' in sys.platform:
    _libcublas_libname_list = ['libcublas.so'] + \
                              ['libcublas.so.%s' % v for v in _linux_version_list]
elif sys.platform == 'darwin':
    _libcublas_libname_list = ['libcublas.dylib']
elif sys.platform == 'win32':
    if sys.maxsize > 2**32:
        _libcublas_libname_list = ['cublas.dll'] + \
            ['cublas64_%s.dll' % v for v in _win32_version_list]
    else:
        _libcublas_libname_list = ['cublas.dll'] + \
            ['cublas32_%s.dll' % v for v in _win32_version_list]
else:
    raise RuntimeError('unsupported platform')

修改为（因为我是在windows上运行的所以只改了一部分）：

if 'linux' in sys.platform:
    _libcublas_libname_list = ['libcublas.so'] + \
                              ['libcublas.so.%s' % v for v in _linux_version_list]
elif sys.platform == 'darwin':
    _libcublas_libname_list = ['libcublas.dylib']
elif sys.platform == 'win32':
    if sys.maxsize > 2**32:
        _libcublas_libname_list = ['cublas64_10.dll'] + \
            ['cublas64_%s.dll' % v for v in _win32_version_list]
    else:
        _libcublas_libname_list = ['cublas.dll'] + \
            ['cublas32_%s.dll' % v for v in _win32_version_list]
else:
    raise RuntimeError('unsupported platform')