Rešeno: onemogućiti GPU u jupyter notebook-u u tensorflow-u

Posljednje ažuriranje: 09/11/2023

GPU je moćan alat za ubrzavanje računanja na velikim skupovima podataka. Međutim, ponekad može biti potrebno onemogućavanje GPU-a kako bi se poboljšale performanse. Jedan uobičajeni razlog za onemogućavanje GPU-a je kada koristite TensorFlow na laptopu ili desktop računaru sa ograničenim resursima. Onemogućavanje GPU-a također može poboljšati performanse kada koristite TensorFlow na mobilnom uređaju.

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

Prvi red uvozi os modul. Drugi red postavlja varijablu okruženja CUDA_VISIBLE_DEVICES na -1. Ovo govori TensorFlow-u da ne koristi nijedan GPU.

Jupyter Notebook

Jupyter Notebook je web aplikacija koja vam omogućava da kreirate i dijelite dokumente s drugima u okruženju za saradnju. Uključuje funkcije kao što su formatiranje bogatog teksta, uređivanje koda uživo i automatsko generiranje dokumenata.

tenzorski tok

TensorFlow je biblioteka za analizu podataka i mašinsko učenje. Omogućava vam da kreirate prilagođene modele i algoritme kako biste ubrzali obuku tih modela. TensorFlow se može koristiti u Pythonu, što olakšava početak.

GPU

GPU u Pythonu je moćan alat koji se može koristiti za razne zadatke, uključujući grafičku obradu, mašinsko učenje i naučno računarstvo. GPU-ovi su specijalizovani mikroprocesori koji su dizajnirani za grafičko prikazivanje visokih performansi i duboko učenje. Python pruža praktičan omot za CUDA biblioteku, koji olakšava pristup GPU resursima.

Slični postovi: