可以在google colab上安装cupy吗?

Posted

技术标签:

【中文标题】可以在google colab上安装cupy吗?【英文标题】:Is it possible to install cupy on google colab? 【发布时间】:2018-08-14 13:27:19 【问题描述】:

我正在尝试在 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

【问题讨论】:

【参考方案1】:

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


原答案

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

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

【讨论】:

只是一个简短的说明,正如下面@Neerajan Saha 所暗示的那样 - 如果您在 Colab 的 GPU 上运行,则默认情况下您只会获得 CuPy(编辑 -> 笔记本设置 -> 硬件加速器 ->图形处理器)。如果没有,你会发现不先pip安装就无法导入cupy。【参考方案2】:

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

chainer.print_runtime_info()

【讨论】:

以上是关于可以在google colab上安装cupy吗?的主要内容,如果未能解决你的问题,请参考以下文章

我可以将 TensorBoard 与 Google Colab 一起使用吗?

如何在 Google Colab 上安装和使用底图?

Google驱动器与COLAB断开连接

pip 安装了 google.colab,现在无法在 Jupyter 笔记本上运行代码

在哪里可以找到安装在google Colab中的现有软件包

在 google colab Python 上处理大型数组