Skip to content

Instantly share code, notes, and snippets.

@wangjiezhe

wangjiezhe/detect-cuda.py

Created January 6, 2022 04:56
Show Gist options
  • Save wangjiezhe/b4a6451e6f4b15b2375564dfca89106f to your computer and use it in GitHub Desktop.
Save wangjiezhe/b4a6451e6f4b15b2375564dfca89106f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
detect-cuda.py
import torch
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.current_device()
torch.cuda.get_device_name(0)
import tensorflow as tf
tf.config.list_physical_devices('GPU')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment