8

我正在尝试在 google colab 上使用 GPU 运行 chainer。这需要安装cupy，但是我无法正确安装它，因为它在我的colab vm中找不到cuda环境。

错误信息如下...

收集 cupy 下载 cupy-2.4.0.tar.gz (1.7MB) 100% |███████████████████████████████ █| 1.7MB 740kB/s 命令 python setup.py egg_info 的完整输出：cc1plus：警告：命令行选项“-Wstrict-prototypes”对 C/ObjC 有效，但对 C++/tmp/tmpds3ikncy/a.cpp:1:10 无效：致命错误：cublas_v2.h：没有这样的文件或目录#include ^~~~~~~~~~~~~编译终止。选项：{'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 环境变量 ************************************ ****************** Traceback（最近一次调用最后）：文件“”，第 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 extensions = make_extensions(arg_options, compiler, use_cython) 文件“/tmp/pip -build-qu5red9h/cupy/cupy_setup_build.py", line 275, in make_extensions raise Exception('Your CUDA environment is invalid.' Exception: Your CUDA environment is invalid. 请检查上面的错误日志。

----------------------------------------

命令“python setup.py egg_info”失败，错误代码 1 在 /tmp/pip-build-qu5red9h/cupy/

4

2 回答 2

14

编辑：Google Colaboratory 现在默认提供 Chainer、CuPy 和 iDeep。不再需要手动安装。

原答案：

您可以使用轮子来避免自己在 Google Colab 上构建 CuPy。Chainer 官方提供了在 Google Colab 上安装 Chainer/CuPy 的代码片段。

!curl https://colab.chainer.org/install | sh -
于 2018-03-07T03:04:44.287 回答
2

我们不需要在 Google colab 中手动安装 cupy。在编辑下的笔记本设置中，我们可以选择 GPU。如果您已经安装了chainer，您可以通过以下方式确认cupy的可用性：

chainer.print_runtime_info()
于 2019-02-11T06:23:14.693 回答