BinaryNet
BinaryNet copied to clipboard
ImportError: No module named six.moves
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
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
Could you make sure that you are suggesting correct versions of Theano and Pylearn2? Thanks
from six.moves import input, zip as izip
Anybody resolved this issue ?
The environment for this can be quite tricky to set up, very specific versions of some packages are required.
I project I worked on BNN-PYNQ used slightly modified versions of this BinaryNet repo to train the networks. The dockerfile in that repo and instructions are still (at least the last time I checked) compatible with this project.
I'd recommend the dockerfiles, but you can also try to follow the instructions directly as well.