如何使用chainer在google colab上从CPU切换到GPU？答案

【问题标题】：How do I switch from CPU to GPU on google colab using chainer?如何使用chainer在google colab上从CPU切换到GPU？
【发布时间】：2018-12-07 05:45:45
【问题描述】：

我按照 Chainer 文档中的说明进行操作，这导致我在运行代码时出错：

RuntimeErrorTraceback (most recent call last)
<ipython-input-9-ffb21f9880f0> in <module>()
      ...
      6 model = Classifier(CompetitionNetwork(n_units = 64))
----> 7 model.to_gpu()
      ...
RuntimeError: CUDA environment is not correctly set up
(see https://github.com/chainer/chainer#installation).No module named cupy

然后我尝试以多种不同的方式安装 cupy，其中之一是

!apt -y install libcusparse8.0 libnvrtc8.0 libnvtoolsext1
!ln -snf /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so.8.0 /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so
!pip install cupy-cuda80 chainer

在导入 cupy 然后运行我的代码后，它一直给我同样的错误：

RuntimeError: CUDA environment is not correctly set up (see
https://github.com/chainer/chainer#installation).No module named cupy

接下来我尝试使用这个安装 cuda：

!wget https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64 -O cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb

!dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.88-1_amd64.deb

!apt-key add /var/cuda-repo-<version>/7fa2af80.pub

!apt-get update

!apt-get install cuda

这花了很长时间，似乎可以工作，但最后还是给了我同样的错误。

似乎在 Google Colab 的 GPU 上使用 Chainer 非常困难，除非我做错了什么。使用 Tensorflow 就容易多了。有人有在 Google 的 GPU 上使用 Chainer 的经验吗？

【问题讨论】：

您可以按照以下步骤在 colab 中获取 GPU 或 TPU：- Runtime-> 更改 Runtime

标签： python neural-network gpu google-colaboratory chainer

【解决方案1】：

您可能想查看此 Chainer 示例。

https://colab.research.google.com/drive/1SsxHvQdSz23kaVov8yKizVD3_2tkXdZM

【讨论】：

这行得通！我从示例中意识到只能使用 python 3 在 GPU 上运行，但仍然很棒，谢谢！
您还可以使用启用 GPU 的 Colab 在 Python 2 上运行 Chainer。

【解决方案2】：

如果你在 Google Colab 环境中搞砸了，

首先尝试通过从 Runtime 菜单中选择“Restart runtime...”来重新启动 Runtime。
如果不能解决问题，请运行以下代码销毁并重新创建容器（注意：您将丢失在容器上创建的文件），以便您可以从干净状态重新开始。运行命令后，您可能需要等待一分钟，然后刷新浏览器以重新加载笔记本。
```
!kill -9 -1
```

您安装 Chainer/CuPy 的安装步骤是正确的。 https://github.com/kmaehashi/chainer-colab

您不必手动安装 CUDA Toolkit； Colab 容器默认提供。

【讨论】：

【解决方案3】：

我检查了 python2 上的 chainer 是否适用于 google-colab。

https://colab.research.google.com/gist/fiarabbit/a44a8b3ff25afc78849c62c2f75b25dd

我确认 chainer w/python2 在 google-colab 上工作。

关键是您不需要像 kmaehashi 给出的 mnist-example 那样通过 wget 安装 cupy。

另外，别忘了开启 GPU。

【讨论】：