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，但Google Collab中未安装anaconda。

感谢您的帮助。

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上不起作用

1 个答案: