-
Notifications
You must be signed in to change notification settings - Fork 222
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Is it possible to run on a normal docker not NVIDIA docker? #2
Comments
I've tried to run the Notebook on my Macbook pro directly (without docker). I've got an error that seems related to the data generator. (TensorFlow version: 1.3.0, Keras version: 2.0.9, Python 3.6.0a2). I am not sure that error would occur on a GPU. What do you think? Layer (type) Output Shape Param # 1 Connected to ================================================================================================== conv1 (Conv2D) (None, 128, 64, 16) 160 the_input[0][0] max1 (MaxPooling2D) (None, 64, 32, 16) 0 conv1[0][0] conv2 (Conv2D) (None, 64, 32, 16) 2320 max1[0][0] max2 (MaxPooling2D) (None, 32, 16, 16) 0 conv2[0][0] reshape (Reshape) (None, 32, 256) 0 max2[0][0] dense1 (Dense) (None, 32, 32) 8224 reshape[0][0] gru1 (GRU) (None, 32, 512) 837120 dense1[0][0] gru1_b (GRU) (None, 32, 512) 837120 dense1[0][0] add_41 (Add) (None, 32, 512) 0 gru1[0][0] gru2 (GRU) (None, 32, 512) 1574400 add_41[0][0] gru2_b (GRU) (None, 32, 512) 1574400 add_41[0][0] concatenate_41 (Concatenate) (None, 32, 1024) 0 gru2[0][0] dense2 (Dense) (None, 32, 23) 23575 concatenate_41[0][0] softmax (Activation) (None, 32, 23) 0 dense2[0][0] ================================================================================================== Epoch 1/1 StopIteration Traceback (most recent call last) in train(img_w, load) /Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/keras/legacy/interfaces.py in wrapper(*args, **kwargs) /Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/keras/engine/training.py in fit_generator(self, generator, steps_per_epoch, epochs, verbose, callbacks, validation_data, validation_steps, class_weight, max_queue_size, workers, use_multiprocessing, shuffle, initial_epoch) StopIteration: |
Current docker image is working only with GPU. You have to change dockerfile and build new image. After that it will be possible to run it on Mac. We had some positive experience to do it. You have to find command "pip install tensorflow-gpu" and change it to "pip install tensorflow". As i remember - |
I'll try it as you suggested. |
Your advice enabled me to run the docker image on my Mac but I still got the same StopIteration error during the training. The StopIteration problem should come from the generator. Any idea? My configuration: TensorFlow version: 1.4.0, Keras version: 2.0.9, Python: 3.6.0a2 Spasibo! Claude Exception in thread Thread-5: StopIteration Traceback (most recent call last) in train(img_w, load) /src/keras/legacy/interfaces.py in wrapper(*args, **kwargs) /src/keras/engine/training.py in fit_generator(self, generator, steps_per_epoch, epochs, verbose, callbacks, validation_data, validation_steps, class_weight, max_queue_size, workers, use_multiprocessing, shuffle, initial_epoch) StopIteration: |
Has anyone been able to find a solution for this issue? I ran into the same error as @ClaudeCoulombe when trying to run this tutorial on MacOS, and changing
|
I use env with anaconda. create python 3.6, tensorflow-gpu 1.8.0 and keras 2.1.5 and it worked for me. |
Nice tutorial, but I've got problem with the docker image.
The run.sh starts with nvidia-docker and I suspect that your docker image should be dependent on NVIDIA libraries and GPUs. Is-it possible to run your docker image on a CPU? More precisely I would like to run the docker image on a Macbook pro. Did you build a compatible docker image?
The text was updated successfully, but these errors were encountered: