首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Cupy找不到CUDA储存库

Cupy找不到CUDA储存库
EN

Stack Overflow用户
提问于 2020-04-28 07:35:29
回答 1查看 5.6K关注 0票数 1

我正在尝试使用cupy,我已经安装了CUDA 10.2和相应版本的cupy,但是当我尝试运行我的代码时,我得到了这个错误:

代码语言:javascript
运行
复制
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.
EN

回答 1

Stack Overflow用户

发布于 2020-04-30 06:15:43

请确保正确设置了CUDA_PATH环境变量(例如,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)。一般来说,它应该由CUDA安装程序自动设置。

CuPy按以下顺序查找数据自动化系统路径。

  1. CUDA_PATH环境变量。
  2. nvcc命令的父目录。variable.
  3. /usr/local/cuda (不适用于CuPy )

中的PATH环境中的每个目录中的nvcc命令

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

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/61474946

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档