User 3122 | 1/25/2016, 7:35:21 AM
I'm trying to use the GPU with graphlab create on an aws amazon linux g2.2xlarge that was built with the Nvidia AMI that had drivers included.
I installed graphlab then ran the following code to upgrade to the gpu acceleration on aws.
pip install --upgrade --no-cache-dir http://static.dato.com/files/graphlab-create-gpu/graphlab-create-1.8.gpu.tar.gz
Then I rebooted the machine just to be safe. However, if I run the following code to test if it is hitting the GPU I can see in the progress logs that it is NOT. What am I doing wrong?
data = graphlab.SFrame('http://s3.amazonaws.com/dato-datasets/mnist/sframe/train') m = graphlab.neuralnet_classifier.create(data, target='label')
PROGRESS: Computing mean image... PROGRESS: Done computing mean image. PROGRESS: Creating neuralnet using cpu