Cupy не может найти хранилище CUDA

Question

Я пытаюсь использовать 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.

Answer 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 -с-заказ установка CUDA