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

Question

我想在google colab上用GPU运行chainer.这需要安装,但我无法正确安装,因为它无法在我的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环境变量**************************************************追溯(最近一次调用最后一次):文件"",第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",第275行,在make_extensions中引发异常('您的CUDA环境无效.'例外:您的CUDA环境无效 .请检查上面的错误日志.

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

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

Answer 1

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

原答案:

您可以使用方向盘来避免在Google Colab上自行构建CuPy.Chainer正式提供片段在Google Colab上安装Chainer/CuPy.

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