Instalar tensorflow
conda instalar tensorflow-gpu
¿La instalación de tensorflow se realizó correctamente?
pip show tensorflow
D: \ ProgramData \ Anaconda3> pip show tensorflow
Nombre: tensorflow
Versión: 2.1.0
Resumen: TensorFlow es un marco de trabajo de aprendizaje automático de código abierto para todos.
Página de inicio: https://www.tensorflow.org/
Autor: Google Inc.
Autor-correo electrónico: [email protected]
Licencia: Apache 2.0
Ubicación: d: \ programdata \ anaconda3 \ lib \ site-packages
Requiere: tensorboard, keras-preprocessing, numpy, protobuf, scipy, opt-einsum, six, google-pasta, keras-applications, absl-py, termcolor, astor, grpcio, tensorflow-estimator, gast, wrapt, wheel
Requerido-por:
D: \ ProgramData \ Anaconda3>
Prueba el código de tensforflow
importar tensorflow como tf
import os
os.environ ['TF_CPP_MIN_LOG_LEVEL'] = '2' # No mostrar mensajes de aviso por debajo del nivel 2
print ('GPU', tf.test.is_gpu_available ())
a = tf.constant (2.0)
b = tf.constant (4.0)
imprimir (a + b)
>>> importar tensorflow como tf
>>> importar os
>>>
>>> os.environ ['TF_CPP_MIN_LOG_LEVEL'] = '2' # 不 显示 等级 2 以下 的 提示 信息
>>>
>>> print ('GPU' , tf.test.is_gpu_available ())
ADVERTENCIA: tensorflow: From <stdin>: 1: is_gpu_available (from tensorflow.python.framework.test_util) está obsoleto y se eliminará en una versión futura.
Instrucciones para la actualización:
Utilice `tf.config.list_physical_devices ('GPU')` en su lugar.
2021-01-01 08: 48: 58.640068: I tensorflow / core / common_runtime / gpu / gpu_device.cc: 1555] Dispositivo encontrado 0 con propiedades:
pciBusID: 0000: 01: 00.0 nombre: GeForce RTX 2060 computeCapability: 7.5
coreClock: 1.71 Núcleo de GHz Cantidad: 30 dispositivos Tamaño de la memoria: 6.
2021-01-01 08: 48: 58.642961: I tensorflow / stream_executor / platform / default / dso_loader.cc: 44] Biblioteca dinámica abierta con éxito cudart64_101.dll
2021-01-01 08: 48: 58.644386: I tensorflow / stream_executor / platform /default/dso_loader.cc:44] Biblioteca dinámica abierta con éxito cublas64_10.dll
2021-01-01 08: 48: 58.645777: I tensorflow / stream_executor / platform / default / dso_loader.cc: 44] Biblioteca dinámica abierta con éxito cufft64_10.dll
2021 -01-01 08: 48: 58.648105: I tensorflow / stream_executor / platform / default / dso_loader.cc: 44] Biblioteca dinámica abierta con éxito curand64_10.dll
2021-01-01 08: 48: 58.649496: I tensorflow / stream_executor / platform / default / dso_loader.cc: 44] Se abrió correctamente la biblioteca dinámica cusolver64_10.dll
2021-01-01 08: 48: 58.650895: I tensorflow / stream_executor / platform / default / dso_loader.cc: 44] Biblioteca dinámica abierta con éxito cusparse64_10.dll
2021-01-01 08: 48: 58.652300: I tensorflow / stream_executor / platform /default/dso_loader.cc:44] Biblioteca dinámica abierta con éxito cudnn64_7.dll
2021-01-01 08: 48: 58.653699: I tensorflow / core / common_runtime / gpu / gpu_device.cc: 1697] Adición de dispositivos gpu visibles: 0
2021- 01-01 08: 48: 58.654885: I tensorflow / core / common_runtime / gpu / gpu_device.cc: 1096] Interconexión de dispositivos StreamExecutor con matriz de borde de fuerza 1:
2021-01-01 08: 48: 58.656959: I tensorflow / core / common_runtime /gpu/gpu_device.cc:1102] 0
2021-01-01 08: 48: 58.657929: I tensorflow / core / common_runtime / gpu / gpu_device.cc: 1115] 0: N
2021-01-01 08: 48: 58.659003: I tensorflow / core / common_runtime / gpu / gpu_device.cc: 1241] Se creó el dispositivo TensorFlow (/ dispositivo: GPU: 0 con 4604 MB de memoria) -> GPU física (dispositivo: 0, nombre: GeForce RTX 2060, ID de bus pci: 0000: 01: 00.0, capacidad de cálculo: 7.5)
GPU Verdadero
>>>
>>> a = tf.constant (2.0)
>>> b = tf.constant (4.0)
>> > print (a + b)
tf.Tensor (6.0, forma = (), dtype = float32)
Empiecen juntos un nuevo estudio