Cupy 找不到 CUDA 存储库

Posted 2023-02-16

【中文标题】Cupy 找不到 CUDA 存储库

【英文标题】：Cupy unable to find CUDA repository

【发布时间】：2020-04-28 07:35:29

【问题描述】：

我正在尝试使用 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.

【问题讨论】：

能否提供导致此错误的代码？

【参考方案1】：

请确保 CUDA_PATH 环境变量设置正确（例如，C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2）。通常它应该由 CUDA 安装程序自动设置。

CuPy 按以下顺序发现 CUDA 路径。

CUDA_PATH 环境变量。 nvcc 命令的父目录。 CuPy 在PATH 环境变量中设置的每个目录中查找nvcc 命令。 /usr/local/cuda（不适用于 Windows）

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