Skip to content

Instantly share code, notes, and snippets.

@nestorandrespe

nestorandrespe/cuda-tensorflow-check.md

Last active September 9, 2021 23:57
Show Gist options
  • Save nestorandrespe/c7370cfb5c34a713171f694bac3026f8 to your computer and use it in GitHub Desktop.
Save nestorandrespe/c7370cfb5c34a713171f694bac3026f8 to your computer and use it in GitHub Desktop.
Download ZIP
Comandos para probar instalación de CUDA
Raw
cuda-tensorflow-check.md

Ejecutar en la consola

pip install tensorflow-gpu

Ejecutar en la consola python

import tensorflow as tf
tf.test.is_built_with_cuda()
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment