Tensorflow安装gpu版本

python
1
pip install tensorflow-gpu==2.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

TensorFlow测试gpu是否可用(2023.08.12)

难度：easy

python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import tensorflow as tf
gpu_device_name = tf.test.gpu_device_name()
print(gpu_device_name)
print(tf.test.is_gpu_available())
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']
print(tf.__version__) 
print(tf.__path__)

@tensorflow测试gpu是否可用