You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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:
Start GPU runtime.
launch jupyter notebook ( with sagemaker-distribution kernel )
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:
TensorFlow can't recognize GPU, and is not even built with CUDA.
Expected behavior
Screenshots
If applicable, add screenshots to help explain your problem.
Desktop (please complete the following information):
The text was updated successfully, but these errors were encountered: