How to install CUDA in Google Colab GPU's

前端 未结 7 1671
悲哀的现实
悲哀的现实 2020-12-05 13:31

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 Cola

7条回答
  •  一向
    一向 (楼主)
    2020-12-05 14:01

    To run in Colab, you need CUDA 8 (mxnet 1.1.0 for cuda 9+ is broken). But Google Colab runs now 9.2. There is, however the way to uninstall 9.2, install 8.0 and then install mxnet 1.1.0 cu80.

    The complete jupyter code is here : Medium

提交回复
热议问题