Skip to content
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

tensorflow is not built with CUDA #275

Open
yang-kaist opened this issue Jul 15, 2024 · 0 comments
Open

tensorflow is not built with CUDA #275

yang-kaist opened this issue Jul 15, 2024 · 0 comments
Labels
bug Something isn't working

Comments

@yang-kaist
Copy link

yang-kaist commented Jul 15, 2024

Describe the bug
Even with GPU instance, tensorflow ( from sagemaker-distribution conda environment ) can't recognize the GPU.
nvidia-smi show I'm using GPU instance, and pytorch can recognize the GPU.

To Reproduce
Steps to reproduce the behavior:

  1. Start GPU runtime.
  2. launch jupyter notebook ( with sagemaker-distribution kernel )
  3. type
import tensorflow as tf
import torch

tf.config.list_physical_devices('GPU') # []
tf.test.is_built_with_cuda() # False
torch.cuda.is_available() # True
  1. See error
    TensorFlow can't recognize GPU, and is not even built with CUDA.

Expected behavior

tf.config.list_physical_devices('GPU') # at least one GPU should be in the list
tf.test.is_built_with_cuda() # True

Screenshots
If applicable, add screenshots to help explain your problem.
image

Desktop (please complete the following information):

  • OS: windows 11
  • Browser : chrome
  • Version : 126.0.6478.127 (Official Build) (64-bit)
@yang-kaist yang-kaist added the bug Something isn't working label Jul 15, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant