如何在Google Colab GPU的CUDA中安装CUDA [英] How to install CUDA in Google Colab GPU's

问题描述

似乎Google Colab GPU并未随附CUDA Toolkit，我如何在CUDA GPU中安装CUDA.在Google Colab中安装mxnet时遇到此错误.

It seems that Google Colab GPU's doesn't come with CUDA Toolkit, how can I install CUDA in Google Colab GPU's. I am getting this error in installing mxnet in Google Colab.

Installing collected packages: mxnet
Successfully installed mxnet-1.2.0

错误:未完全安装，无法利用GPU进行计算. 请确保您已安装CUDA并在其中运行以下行 您的终端，然后重试:

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

根据您的CUDA版本调整"cu90"("cu75"和"cu80"为 也提供). 您也可以通过调用turicreate.config.set_num_gpus(0)完全禁用GPU使用. 发生异常，请使用％tb查看完整的追溯.

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

推荐答案

我几乎相信Google Colab已预先安装了Cuda ...您可以通过打开一个新笔记本并键入!nvcc --version来确保这将返回已安装Cuda版本.

I pretty much believe that Google Colab has Cuda pre-installed... You can make sure by opening a new notebook and type !nvcc --version which would return the installed Cuda version.

这是我的:

这篇关于如何在Google Colab GPU的CUDA中安装CUDA的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！