Regarding GPU Integration issue

awslabs / handwritten-text-recognition-for-apache-mxnet

This repository lets you train neural networks models for performing end-to-end full-page handwriting recognition using the Apache MXNet deep learning frameworks on the IAM Dataset.

Apache License 2.0

481 stars 189 forks source link

Regarding GPU Integration issue #29

Closed Neelesh1121 closed 4 years ago

Neelesh1121 commented 4 years ago

When I am executing "ctx = mx.gpu() if mx.context.num_gpus() > 0 else mx.cpu()"
The output of ctx is coming only cpu() and the output of "mx.context.num_gpus()" is coming only 0 even I have enabled the GPU. I am using Mxnet 1.4.0 and leven==1.0.4.

jonomon commented 4 years ago

Hi Neelesh, did you get the GPU version of MXNet? e.g., pip install mxnet-cu102 as per https://mxnet.apache.org/get_started?platform=linux&language=python&processor=gpu&environ=pip&

Neelesh1121 commented 4 years ago

Hi Jonomon, Thanks for your quick reply, Yes, I solved this problem by installing the GPU version of MXnet