Ce que je suis entré lors de l'utilisation de Tensorflow-gpu

J'étais accro à l'apprentissage automatique en utilisant tensorflow, je vais donc le décrire.

environnement

environnement version
OS Windows
tensorflow 2.3.0
CUDA 11.0

problème

import tensorflow as tf
mnist = tf.keras.datasets.mnist

(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(512, activation=tf.nn.relu),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)

Lorsque vous essayez d'exécuter du code comme celui ci-dessus

F .\tensorflow/core/kernels/random_op_gpu.h:232] Non-OK-status: GpuLaunchKernel(FillPhiloxRandomKernelLaunch<Distribution>, num_blocks, block_size, 0, d.stream(), gen, data, size, dist) status: Internal: invalid configuration argument

Une erreur s'est produite et il est devenu impossible d'exécuter.

Cette erreur

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

Il était possible de l'éviter en le paramétrant pour ne pas utiliser de GPU. Je pensais que c'était causé par CUDA et j'ai essayé de télécharger à nouveau et de revoir les paramètres du chemin, mais le résultat n'a pas changé ...

Solution

pip install tf-nightly-gpu

J'ai pu le résoudre. Je me suis demandé ce que c'était tf-nightly-gpu, mais il semblait être plus récent que la version 2.3.0 dans la dernière version de tensorflow.

pip list

Quand je regarde dans

tf-estimator-nightly     2.4.0.dev2020091501
tf-nightly-gpu           2.4.0.dev20200912

Il semblait en être au stade de développement.

Je ne savais pas pourquoi cela résolvait l'erreur ...

référence

https://itips.krsw.biz/tensorflow-keras-gpu-deactivate/ https://github.com/tensorflow/tensorflow/issues/30665

Recommended Posts

Ce que je suis entré lors de l'utilisation de Tensorflow-gpu
5 raisons pour lesquelles je suis entré dans Python
J'ai eu une erreur lors de l'utilisation de Tensorboard avec Pytorch
Ce à quoi j'étais accro lors de l'utilisation de Python tornado
J'ai eu une erreur de valeur lors de l'utilisation de JUMAN ++ avec PyKNP
Qu'est-ce qui a été demandé lors de l'utilisation de Random Forest dans la pratique
Ce que je suis tombé sur l'utilisation d'Airflow
J'ai un TypeError: l'objet 'int' n'est pas itérable lors de l'utilisation de keras
Scribble ce que j'ai utilisé lors de l'utilisation d'ipython dans le formatage des données de position
Ce sur quoi je suis tombé lors de l'utilisation de CodeIgniter sur un serveur Linux
Ce à quoi j'ai fait référence en étudiant tkinter
Ce que Rubyist doit savoir lors de l'utilisation de pyenv
Ce que j'ai obtenu de Python Boot Camp
Un rappel de ce que je suis resté coincé lors du démarrage d'Atcoder avec python
[Ansible] Ce à quoi je fais attention lorsque j'écris ansible
Ce que j'ai fait quand je suis resté coincé dans le délai avec lambda python
Ce que j'ai fait lors de la mise à jour de Python 2.6 vers 2.7
[Question] Que se passe-t-il si vous utilisez% en python?
Linux Ubuntu16.04 J'ai eu une erreur effrayante lorsque j'ai exécuté une commande en utilisant sudo
Qu'utilisez-vous lorsque vous testez avec Python?
Ce que j'ai appris sur l'IA / l'apprentissage automatique avec Python (1)
Quand j'ai essayé d'introduire python3 dans atom, je suis resté coincé
J'ai un UnicodeDecodeError lorsque je pip installé sur ubuntu
Ce que j'ai appris sur l'IA / l'apprentissage automatique avec Python (3)
Ce que j'ai appris sur l'IA / l'apprentissage automatique avec Python (2)
J'ai essayé d'utiliser paramétré
J'ai essayé d'utiliser argparse
J'ai essayé d'utiliser la mimesis
Une note de ce que j'ai appris quand j'ai pensé à utiliser pyenv ou virtualenv sous Windows
J'ai essayé d'utiliser anytree
J'ai essayé d'utiliser aiomysql
J'ai essayé d'utiliser Summpy
J'ai obtenu "ModuleNotFoundError: aucun module nommé" azure "" lors de l'exécution d'un programme à l'aide du SDK Azure pour Python
J'ai essayé d'utiliser coturn
J'ai essayé d'utiliser Pipenv
J'ai essayé d'utiliser matplotlib
J'ai essayé d'utiliser "Anvil".
J'ai essayé d'utiliser Hubot
J'ai essayé d'utiliser ESPCN
J'ai essayé d'utiliser openpyxl
J'ai essayé d'utiliser Ipython
J'ai essayé d'utiliser PyCaret
J'ai essayé d'utiliser cron
J'ai essayé d'utiliser ngrok
J'ai essayé d'utiliser face_recognition
J'ai essayé d'utiliser Jupyter
J'ai un sqlite3.OperationalError
J'ai essayé d'utiliser doctest
Précautions lors de l'utilisation de Chainer
J'ai essayé d'utiliser du folium
J'ai essayé d'utiliser jinja2
J'ai essayé d'utiliser du folium
J'ai essayé d'utiliser la fenêtre de temps
Ce à quoi j'étais accro lorsque j'ai construit mon propre réseau de neurones en utilisant les poids et les biais que j'ai obtenus avec le classificateur MLP de scikit-learn.
Éléments à prendre en compte lors de la mise en œuvre d'Airflow avec docker-compose
Quand j'ai rendu CaboCha disponible en python3, je suis resté bloqué (Windows 10)
Ce que je suis resté coincé autour de l'interface graphique dans l'environnement python WSL
Ce que j'ai appris sur l'IA / l'apprentissage automatique avec Python (4)