可以在谷歌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 -

Answer 2

我們不需要在Google colab中手動安裝cupy。 在Edit下的Notebook Settings中，我們可以選擇GPU。 如果您已經安裝了chainer，您可以通過以下方式確認可用性：

chainer.print_runtime_info()