Ошибка Google Colab "для работы требуется графический процессор" на Theano / Lasagne - PullRequest
Новая
       134

Ошибка Google Colab "для работы требуется графический процессор" на Theano / Lasagne

0 голосов
/ 12 июля 2020

При запуске моего python скрипта в Google Colab с использованием Theano и Lasagne я получаю следующую ошибку графического процессора.

Когда я запускаю: 

! THEANO_FLAGS='device=cuda' python ./score.py

Я получаю: 

Using device cuda: Tesla P4
[nltk_data] Downloading package punkt to /root/nltk_data...
[nltk_data]   Package punkt is already up-to-date!
Traceback (most recent call last):
  File "./score.py", line 2, in <module>
    import generate 
  File "/content/gdrive/My Drive/storygen/source/generate.py", line 18, in <module>
    from lasagne.layers.corrmm import Conv2DMMLayer as ConvLayer
  File "/usr/local/lib/python3.6/site-packages/lasagne/layers/corrmm.py", line 22, in <module>
    raise ImportError("requires a GPU to work")  # pragma: no cover
ImportError: requires a GPU to work

Я включил графический процессор в моей среде выполнения и перезапустил его, а также проверил, работает ли графический процессор (как видно выше, Tesla P4).

Я использую следующие версии для пакетов: 

Lasagne                              0.1
Theano                               0.7.0

Вот script.py : 

import os, json, base64
import generate

generatae.py : 

import io, cv2, base64
import pickle as pkl
import numpy, copy, sys, skimage.transform
import config
import nltk
import lasagne
from lasagne.layers import InputLayer, DenseLayer, NonlinearityLayer, DropoutLayer, MaxPool2DLayer as PoolLayer 
from lasagne.nonlinearities import softmax
from lasagne.utils import floatX
if not config.FLAG_CPU_MODE:
    from lasagne.layers.corrmm import Conv2DMMLayer as ConvLayer
...