Theano 支持 GPU 不适用于 google Colab

Question

我正在尝试训练基于 Theano 库的模型。由于我的计算机没有足够的内存来训练这个模型，我需要在 Google Colab 中进行训练。但是我无法激活Theano的GPU支持。

这就是我安装 Theano 和 Lasagne 的方式

!pip install --upgrade https://github.com/Theano/Theano/archive/master.zip
!pip install --upgrade https://github.com/Lasagne/Lasagne/archive/master.zip

这是我尝试激活 GPU 支持的方法

import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=cuda,floatX=float32"
import theano
print(theano.config.__getattribute__('device'))

ERROR (theano.gpuarray): pygpu was configured but could not be imported or is too old (version 0.7 or higher required)
NoneType: None

cuda

我尝试使用 anaconda 升级 pygpu，但 anaconda 未安装在 Google Collab 中。

提前感谢您的帮助。

Answer 1

#install conda
!wget -c https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh

!chmod +x Anaconda3-5.1.0-Linux-x86_64.sh

!bash ./Anaconda3-5.1.0-Linux-x86_64.sh -b -f -p /usr/local

import sys

sys.path.append('/usr/local/lib/python3.6/site-packages/')

#install theano and pygpu
!conda install theano pygpu

一切顺利...

Theano 支持 GPU 不适用于 google Colab

Theano support GPU is not working on google Colab

gpu

machine-learning

theano

google-colaboratory