Skip to content

Instantly share code, notes, and snippets.

@hsleonis

hsleonis/cuda_version.py

Created November 30, 2021 10:39
Show Gist options
  • Save hsleonis/a6614288e29404fd9d0b8ed9deec3e9e to your computer and use it in GitHub Desktop.
Save hsleonis/a6614288e29404fd9d0b8ed9deec3e9e to your computer and use it in GitHub Desktop.
Download ZIP
Get CUDA Version
Raw
cuda_version.py
import tensorflow as tf
from tensorflow.python.platform import build_info as build
print(f"tensorflow version: {tf.__version__}")
print(f"Cuda Version: {build.build_info['cuda_version']}")
print(f"Cudnn version: {build.build_info['cudnn_version']}")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment