CuPy - ошибка импорта - nvidia lib. не найдено

Я получаю следующую ошибку при попытке импортировать CuPy:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python2.7/dist-packages/cupy/__init__.py", line 20, in <module>
    six.reraise(ImportError, ImportError(msg), exc_info[2])
  File "/usr/local/lib/python2.7/dist-packages/cupy/__init__.py", line 9, in <module>
    from cupy import core  # NOQA
  File "/usr/local/lib/python2.7/dist-packages/cupy/core/__init__.py", line 1, in <module>
    from cupy.core import core  # NOQA
  File "cupy/core/core.pyx", line 1, in init cupy.core.core (cupy/core/core.cpp:103472)
  File "/usr/local/lib/python2.7/dist-packages/cupy/cuda/__init__.py", line 3, in <module>
    from cupy.cuda import compiler  # NOQA
  File "/usr/local/lib/python2.7/dist-packages/cupy/cuda/compiler.py", line 12, in <module>
    from cupy.cuda import function
  File "cupy/cuda/function.pyx", line 1, in init cupy.cuda.function (cupy/cuda/function.cpp:6346)
ImportError: CuPy is not correctly installed. Please check your environment, uninstall CuPy and reinstall it with `pip install cupy --no-cache-dir -vvvv`.

original error: libnvidia-fatbinaryloader.so.375.82: cannot open shared object file: No such file or directory

Я попытался переустановить CuPy с предложением. Также я попытался скопировать файл libnvidia-fatbinaryloader.so.375.82 из /usr/lib/nvidia-375 в /usr/local/cuda-8.0/lib64/, но не работает.

Фактически, когда я копирую этот файл в /usr/local/cuda-8.0/lib64/, я получаю эту ошибку (при попытке создать массив):

>>> import cupy as cp

>>> x_gpu = cp.array([1, 2, 3])

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python2.7/dist-packages/cupy/creation/from_data.py", line 26, in array
    return core.array(obj, dtype, copy, ndmin)
  File "cupy/core/core.pyx", line 1822, in cupy.core.core.array (cupy/core/core.cpp:59154)
  File "cupy/core/core.pyx", line 1848, in cupy.core.core.array (cupy/core/core.cpp:58918)
  File "cupy/core/core.pyx", line 92, in cupy.core.core.ndarray.__init__ (cupy/core/core.cpp:6509)
  File "cupy/cuda/memory.pyx", line 283, in cupy.cuda.memory.alloc (cupy/cuda/memory.cpp:5911)
  File "cupy/cuda/memory.pyx", line 263, in cupy.cuda.memory._malloc (cupy/cuda/memory.cpp:5853)
  File "cupy/cuda/memory.pyx", line 264, in cupy.cuda.memory._malloc (cupy/cuda/memory.cpp:5739)
  File "cupy/cuda/memory.pyx", line 34, in cupy.cuda.memory.Memory.__init__ (cupy/cuda/memory.cpp:1789)
  File "cupy/cuda/device.pyx", line 67, in cupy.cuda.device.Device.__init__ (cupy/cuda/device.cpp:1789)
  File "cupy/cuda/runtime.pyx", line 158, in cupy.cuda.runtime.getDevice (cupy/cuda/runtime.cpp:2563)
  File "cupy/cuda/runtime.pyx", line 130, in cupy.cuda.runtime.check_status (cupy/cuda/runtime.cpp:2242)
cupy.cuda.runtime.CUDARuntimeError: cudaErrorUnknown: unknown error

Моя версия CUDA: 8.0

Ubuntu 16.04

Python: 2,7

Может ли кто-нибудь помочь? Я хочу избежать переустановки CUDA, если это возможно. Если вам нужна дополнительная информация для ее решения, не стесняйтесь спрашивать. Спасибо!

1 ответ

Я переустановил драйверы NVIDIA, и это сработало..

Другие вопросы по тегам