-
Notifications
You must be signed in to change notification settings - Fork 339
Description
I installed pylearn2 and theano(bleeding-edge version) following the links given on the README.
When running mnist.py I encountered the following error:
Using cuDNN version 5005 on context None
Preallocating 3027/4036 Mb (0.750000) on cuda
Mapped name None to device cuda: GeForce GTX 970 (0000:0F:00.0)
Traceback (most recent call last):
File "mnist.py", line 24, in
from pylearn2.datasets.mnist import MNIST
File "/home/jiyu/fyp/pylearn2/pylearn2/init.py", line 4, in
from pylearn2.utils.logger import configure_custom
File "/home/jiyu/fyp/pylearn2/pylearn2/utils/init.py", line 11, in
from theano.compat.six.moves import input, zip as izip
ImportError: No module named six.moves
I tried to solve the issue by installing Theano 0.8, but then a new problem occurred:
ERROR (theano.sandbox.gpuarray): Could not initialize pygpu, support disabled
Traceback (most recent call last):
File "/home/jiyu/miniconda2/envs/py27/lib/python2.7/site-packages/theano/sandbox/gpuarray/init.py", line 95, in
init_dev(config.device)
File "/home/jiyu/miniconda2/envs/py27/lib/python2.7/site-packages/theano/sandbox/gpuarray/init.py", line 46, in init_dev
"Make sure Theano and libgpuarray/pygpu "
RuntimeError: ('Wrong major API version for gpuarray:', 2, 'Make sure Theano and libgpuarray/pygpu are in sync.')
Could you make sure that you are suggesting correct versions of Theano and Pylearn2?
Thanks