Cupy не может найти репозиторий CUDA

Я пытаюсь использовать cupy, и я установил CUDA 10.2 и соответствующую версию cupy, но я получаю эту ошибку, когда пытаюсь запустить свой код:


Traceback (most recent call last):
  File "C:/Users/lasse/PycharmProjects/AI/main.py", line 2, in <module>
    import cupy as cp
  File "C:\Users\lasse\AppData\Local\Programs\Python\Python38\lib\site-packages\cupy\__init__.py", line 14, in <module>
    _environment._setup_win32_dll_directory()  # NOQA
  File "C:\Users\lasse\AppData\Local\Programs\Python\Python38\lib\site-packages\cupy\_environment.py", line 37, in _setup_win32_dll_directory
    raise RuntimeError('CUDA path could not be detected.')
RuntimeError: CUDA path could not be detected.

person Lanttu1243    schedule 28.04.2020    source источник
comment
Не могли бы вы предоставить код, приводящий к этой ошибке?   -  person Vincent    schedule 28.04.2020


Ответы (1)


Убедитесь, что переменная среды CUDA_PATH установлена ​​правильно (например, C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2). Обычно он должен быть установлен автоматически установщиком CUDA.

CuPy обнаруживает путь CUDA в следующем порядке.

  1. CUDA_PATH переменная среды.
  2. Родительский каталог команды nvcc. CuPy ищет команду nvcc в каждом каталоге, заданном в переменной среды PATH.
  3. /usr/local/cuda (не применимо для Windows)

https://docs-cupy.chainer.org/en/stable/install.html#working-with-custom-cuda-installation

person kmaehashi    schedule 30.04.2020