CODEWITHSUNDEEP
amazon-web-servicestensorflow
Daniil Yefimov
Daniil Yefimov

Reputation: 1012

AWS SageMaker: GPU is not available for Notebook instance

I want to run my code in notebook instance on AWS Sagemaker. My instance is ml.m5.4xlarge

But if I just import TensorFlow in standard way it, then TensorFlow on GPU is not available(only CPU is used):

import tensorflow as tf
tf.config.list_physical_devices('GPU')

['']

What should I do in order to use GPUs for tensorflow?

P.S to be clear I don't need to deploy anything. I just want to train my model using notebook instance(for Kaggle competition)

EDIT: I have used conda_tensorflow2_p36 notebook

Upvotes: 0

Views: 2304

Answers (1)

rok
rok

Reputation: 2785

The instance ml.m5.4xlarge does not have any GPU. You have to choose an instance type with a GPU. Take a look here

Upvotes: 3

Related Questions