可以在google colab上安装cupy吗？

Question

我正在尝试在 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：没有这样的文件或目录 ＃包括 ^~~~~~~~~~~~~ 编译终止。 选项：{'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

Answer 1

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

---

原答案：

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

!curl https://colab.chainer.org/install | sh -

Answer 2

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

chainer.print_runtime_info()