我正在尝试使用 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.
最佳答案
请确保正确设置CUDA_PATH环境变量(例如C:\Program Files\NVIDIA GPUComputing 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
关于python - Cupy 无法找到 CUDA 存储库，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/61474946/