python - 如何在 Google Colab GPU 中安装 CUDA

Question

Google Colab GPU 似乎没有附带 CUDA 工具包，我如何在 Google Colab GPU 中安装 CUDA。我在 Google Colab 中安装 mxnet 时遇到此错误。

Installing collected packages: mxnet
Successfully installed mxnet-1.2.0

ERROR: Incomplete installation for leveraging GPUs for computations. Please make sure you have CUDA installed and run the following line in your terminal and try again:

pip uninstall -y mxnet && pip install mxnet-cu90==1.1.0

Adjust 'cu90' depending on your CUDA version ('cu75' and 'cu80' are also available). You can also disable GPU usage altogether by invoking turicreate.config.set_num_gpus(0). An exception has occurred, use %tb to see the full traceback.

SystemExit: 1

Answer 1

Cuda 未显示在您的笔记本上，因为您尚未在 Colab 中启用 GPU。

Google Colab 带有 GPU 或不带 GPU 两种选项。您可以在运行时设置中启用或禁用 GPU

Go to Menu > Runtime > Change runtime.

将硬件加速改为 GPU。

检查GPU是否正在运行，运行以下命令

!nvidia-smi

如果输出类似于下图，则表示您的 GPU 和 cuda 正在运行。您还可以看到 CUDA 版本。

之后检查 PyTorch 是否能够使用 GPU，运行以下代码。

import torch
torch.cuda.is_available()
# Output would be True if Pytorch is using GPU otherwise it would be False.

要检查 TensorFlow 是否能够使用 GPU，请运行以下代码。

import tensorflow as tf
tf.test.gpu_device_name()
# Standard output is '/device:GPU:0'