Поддержка Theano GPU не работает на Google Colab - PullRequest
Новая
       107

Поддержка Theano GPU не работает на Google Colab

0 голосов
/ 30 октября 2019

Я пытаюсь тренировать модель, основанную на библиотеке Theano. Поскольку моему компьютеру не хватает памяти для обучения этой модели, мне нужно сделать это в Google Colab. Но я не могу активировать поддержку GPU в Theano.

Вот так я установил 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

Я пытался обновить pygpu с помощью anaconda, но anaconda не установлена ​​в Google Collab.

Спасибо за помощь.

...