pyotrch检验命令

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

tensorflow检验命令

import tensorflow as tf
print(tf.__version__)
tf.test.is_gpu_available()