废话不多说,直接上代码。
'''
tensorflow2
# '''
# import tensorflow as tf
# import os
# tf.test.is_gpu_available()
# tf.config.list_physical_devices('GPU')
# tf.test.gpu_device_name()
# import tensorflow as tf
# print(tf.config.list_physical_devices('GPU'))
'''tensorflow1'''
# import tensorflow as tf
# hello=tf.constant('hello,world')
# sess=tf.Session()
# print(sess.run(hello))
'''
pytorch
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
'''
import torch
import torchvision
print(torch.cuda.is_available())
# import os
# from tensorflow.python.client import device_lib
#
# os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
#
# # if __name__ == "__main__":
# print(device_lib.list_local_devices())
'''
mxnet
'''
# from mxnet import nd
# x = nd.array([[1,2],[3,4]])
# print(x)
# y = nd.array([[5,6],[7,8]])
# print(y)
# import tensorflow as tf
#
# gpu_device_name = tf.test.gpu_device_name()
# tf.test.is_gpu_available()
# print(gpu_device_name)
# from tensorflow.python.client import device_lib
#
# # 列出所有的本地机器设备
# local_device_protos = device_lib.list_local_devices()
# # 打印
# # print(local_device_protos)
#
# # 只打印GPU设备
# [print(x) for x in local_device_protos if x.device_type == 'GPU']
'''paddle'''
# import paddle
# paddle.utils.run_check()