我正在尝试在 google colab 上使用 GPU 运行 chainer。这需要安装 cupy，但是我无法正确安装它，因为它在我的 colab 虚拟机中找不到 cuda 环境。
错误信息如下...
收集丘比
下载cupy-2.4.0.tar.gz (1.7MB)
100% |████████████████████████████████| 1.7MB 740kB/秒
命令 python setup.py Egg_info 的完整输出：
cc1plus：警告：命令行选项“-Wstrict-prototypes”对 C/ObjC 有效，但对 C++ 无效
/tmp/tmpds3ikncy/a.cpp:1:10：致命错误：cublas_v2.h：没有这样的文件或目录
＃包括
^~~~~~~~~~~~~
编译终止。
选项：{'profile'：False，'linetrace'：False，'annotate'：False，'no_cuda'：False}
******************************************************* 警告：nvcc 不在路径中。
*警告：请将路径设置为 nvcc。
******************************************************
包含目录：[]
库目录：[]
命令“x86_64-linux-gnu-gcc”失败，退出状态为 1
******************************************************* 警告：包含未找到的文件：['cublas_v2.h'、'cuda.h'、'cuda_profiler_api.h'、'cuda_runtime.h'、'curand.h'、'cusparse.h'、'nvrtc.h' , 'nvToolsExt.h']
*警告：跳过安装 cuda 支持
*** 警告：检查您的 CFLAGS 环境变量
******************************************************
回溯（最近一次调用最后一次）：
文件“”，第 1 行，位于
文件“/tmp/pip-build-qu5red9h/cupy/setup.py”，第 32 行，位于
ext_modules = cupy_setup_build.get_ext_modules()
文件“/tmp/pip-build-qu5red9h/cupy/cupy_setup_build.py”，第 385 行，在 get_ext_modules 中
扩展= make_extensions（arg_options，编译器，use_cython）
文件“/tmp/pip-build-qu5red9h/cupy/cupy_setup_build.py”，第 275 行，在 make_extensions 中
raise Exception('您的 CUDA 环境无效。'
异常：您的 CUDA 环境无效。请检查上面的错误日志。
----------------------------------------
命令“python setup.py Egg_info”失败，/tmp/pip-build-qu5red9h/cupy/ 中的错误代码为 1
Edit：Google Colaboratory 现在默认提供 Chainer、CuPy 和 iDeep。不再需要手动安装。
原答案:
您可以使用轮子来避免在 Google Colab 上自行构建 CuPy。
Chainer官方提供snippet在 Google Colab 上安装 Chainer/CuPy。
!curl https://colab.chainer.org/install | sh -
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
