cupy 安装时提示报错 Exception: Your CUDA environment is invalid.

最新推荐文章于 2025-05-28 09:28:18 发布

xueying_2017

最新推荐文章于 2025-05-28 09:28:18 发布

阅读量5.3k

点赞数 5

CC 4.0 BY-SA版权

分类专栏：科研深度学习

本文链接：https://blog.csdn.net/xueying_2017/article/details/116045080

深度学习同时被 2 个专栏收录

7 篇文章

订阅专栏

科研

5 篇文章

订阅专栏

在尝试使用pip安装CuPy时遇到错误，提示CUDA环境不合法。通过检查CUDA版本（10.2）并指定对应的CuPy版本（cupy-cuda102）进行安装，成功解决了问题。错误信息涉及到cublas_v2.h等文件未找到，但最终安装完成。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

执行 pip install cupy 之后报错，提示cuda环境不合法

解决方法：

1. 查看当前环境cuda版本

cat /usr/local/ cuda /version.txt

得到cuda10.2

2. 根据cuda版本指定cupy版本安装

pip install cupy-cuda102

成功安装

ref:https://blog.csdn.net/qq907482638/article/details/102586724

具体报错信息如下

ERROR: Command errored out with exit status 1:
     command: /usr/local/miniconda3/envs/dl/bin/python -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/setup.py'"'"'; __file__='"'"'/tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(__file__);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' egg_info --egg-base /tmp/pip-pip-egg-info-vzzat06i
         cwd: /tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/
    Complete output (25 lines):
    cc1plus: warning: command line option '-Wstrict-prototypes' is valid for C/ObjC but not for C++
    /tmp/tmp2t32ou6n/a.cpp:1:10: fatal error: cublas_v2.h: No such file or directory
     #include <cublas_v2.h>
              ^~~~~~~~~~~~~
    compilation terminated.
    Options: {'profile': False, 'linetrace': False, 'annotate': False, 'no_cuda': False}
    Include directories: ['/usr/local/cuda/include']
    Library directories: ['/usr/local/cuda/lib64']
    command '/usr/local/miniconda3/envs/dl/bin/x86_64-conda_cos6-linux-gnu-cc' failed with exit status 1
    **************************************************
    *** WARNING: Include files not found: ['cublas_v2.h', 'cuda.h', 'cuda_profiler_api.h', 'cuda_runtime.h', 'curand.h', 'cusparse.h', 'nvrtc.h', 'nvToolsExt.h']
    *** WARNING: Skip installing cuda support
    *** WARNING: Check your CFLAGS environment variable
    **************************************************
    /tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/setup.py:3: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
      import imp
    Traceback (most recent call last):
      File "<string>", line 1, in <module>
      File "/tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/setup.py", line 32, in <module>
        ext_modules = cupy_setup_build.get_ext_modules()
      File "/tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/cupy_setup_build.py", line 374, in get_ext_modules
        extensions = make_extensions(arg_options, compiler, use_cython)
      File "/tmp/pip-install-hi6brf50/cupy_2e41c9f2cfd94922b9669c2acf9fd3a1/cupy_setup_build.py", line 267, in make_extensions
        raise Exception('Your CUDA environment is invalid. '
    Exception: Your CUDA environment is invalid. Please check above error log.
    ----------------------------------------