Question

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

错误信息如下......

收集丰满正在下载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环境变量 ************************************************** Traceback（最近一次调用最后一次）：文件“”，第1行，in 文件“/tmp/pip-build-qu5red9h/cupy/setup.py”，第32行，in ext_modules = cupy_setup_build.get_ext_modules（）在get_ext_modules中输入文件“/tmp/pip-build-qu5red9h/cupy/cupy_setup_build.py”，第385行 extensions = make_extensions（arg_options，compiler，use_cython）在make_extensions中输入文件“/tmp/pip-build-qu5red9h/cupy/cupy_setup_build.py”，第275行提高异常（'你的CUDA环境无效'。例外：您的CUDA环境无效。请检查上面的错误日志。

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

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

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()

可以在谷歌colab上安装cupy吗？

2 个答案: