Heim Backend-Entwicklung Python-Tutorial So verwenden Sie die GPU-Beschleunigung in Python

So verwenden Sie die GPU-Beschleunigung in Python

May 29, 2019 pm 03:33 PM
python

Ich habe viele Tutorials gelesen und die Verwendung von os.environ["CUDA_VISIBLE_DEVICES"] = "1" oder export CUDA_VISIBLE_DEVICES=1 erwähnt.

Aber die Geschwindigkeit hat sich nicht verbessert, als ich es dem Code hinzugefügt habe Es hat lange gedauert, bis ich es gefunden habe. Das Problem ist, dass die beiden oben genannten Codezeilen nur funktionieren, wenn Ihr Computer über mehr als zwei GPUs verfügt!

So verwenden Sie die GPU-Beschleunigung in Python

Da mein Computer nur über eine GPU verfügt, muss ich den Parameter „1“ auf „0“ ändern, da sonst das Gerät „1“ nicht gefunden werden kann standardmäßig verwendet werden und die Geschwindigkeit wird nicht verbessert.

Wenn Sie das Python-Programm im Terminal ausführen, verwenden Sie den Befehl: CUDA_VISIBLE_DEVICES=0 python filename.py

Wenn Sie mehrere Karten haben und mehrere GPU-Trainings angeben möchten, können Sie dies festlegen Es sieht so aus:

def set_gpus(gpu_index):
    if type(gpu_index) == list:
        gpu_index = ','.join(str(_) for _ in gpu_index)
    if type(gpu_index) ==int:
        gpu_index = str(gpu_index)
    os.environ["CUDA_VISIBLE_DEVICES"] = gpu_index

Dies kann durch Aufrufen der Funktion set_gpu erreicht werden.

Beachten Sie unbedingt, dass CUDA_VISIBLE_DEVICES, alle! eins! individuell! Charakter! Mutter! Wenn Sie versehentlich CUDA_VISIBLE_DIVICES oder CUDA_VISIABLE_DEVICES schreiben, meldet der Code keinen Fehler, aber der GPU-Aufruf ist nicht erfolgreich,

Das obige ist der detaillierte Inhalt vonSo verwenden Sie die GPU-Beschleunigung in Python. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!

Erklärung dieser Website
Der Inhalt dieses Artikels wird freiwillig von Internetnutzern beigesteuert und das Urheberrecht liegt beim ursprünglichen Autor. Diese Website übernimmt keine entsprechende rechtliche Verantwortung. Wenn Sie Inhalte finden, bei denen der Verdacht eines Plagiats oder einer Rechtsverletzung besteht, wenden Sie sich bitte an admin@php.cn

Heiße KI -Werkzeuge

Undress AI Tool

Undress AI Tool

Ausziehbilder kostenlos

Undresser.AI Undress

Undresser.AI Undress

KI-gestützte App zum Erstellen realistischer Aktfotos

AI Clothes Remover

AI Clothes Remover

Online-KI-Tool zum Entfernen von Kleidung aus Fotos.

Stock Market GPT

Stock Market GPT

KI-gestützte Anlageforschung für intelligentere Entscheidungen

Heiße Werkzeuge

Notepad++7.3.1

Notepad++7.3.1

Einfach zu bedienender und kostenloser Code-Editor

SublimeText3 chinesische Version

SublimeText3 chinesische Version

Chinesische Version, sehr einfach zu bedienen

Senden Sie Studio 13.0.1

Senden Sie Studio 13.0.1

Leistungsstarke integrierte PHP-Entwicklungsumgebung

Dreamweaver CS6

Dreamweaver CS6

Visuelle Webentwicklungstools

SublimeText3 Mac-Version

SublimeText3 Mac-Version

Codebearbeitungssoftware auf Gottesniveau (SublimeText3)

Heiße Themen

So installieren Sie Pakete aus einer Anforderungen.txt -Datei in Python So installieren Sie Pakete aus einer Anforderungen.txt -Datei in Python Sep 18, 2025 am 04:24 AM

Führen Sie Pipinstall-Rrequirements.txt aus, um das Abhängigkeitspaket zu installieren. Es wird empfohlen, zunächst die virtuelle Umgebung zu erstellen und zu aktivieren, um Konflikte zu vermeiden, sicherzustellen, dass der Dateipfad korrekt ist und dass die PIP aktualisiert wurde, und Optionen wie-No-Deps oder -User, um das Installationsverhalten bei Bedarf anzupassen.

So testen Sie Python -Code mit PyTest So testen Sie Python -Code mit PyTest Sep 20, 2025 am 12:35 AM

Python ist ein einfaches und leistungsstarkes Testwerkzeug in Python. Nach der Installation werden Testdateien automatisch gemäß den Namensregeln ermittelt. Schreiben Sie eine Funktion, die mit Test_ für Assertionstests beginnt, verwenden Sie @PyTest.Fixure, um wiederverwendbare Testdaten zu erstellen, die Ausnahmen über pyTest.raises zu überprüfen, unterstützt die laufenden Tests und mehrere Befehlszeilenoptionen und verbessert die Testeneffizienz.

So behandeln Sie Befehlszeilenargumente in Python So behandeln Sie Befehlszeilenargumente in Python Sep 21, 2025 am 03:49 AM

TheArgParSemoduleiTherecommendedwaytoHandleCommand-Lineargumentesinpython, das Robustparsing, Typevalidation, Helpsages, AndersHandling berücksichtigt; usesys.argvForSimpecaseSeRequiringMinimalsetup.

Von Anfängern bis zu Experten: 10 Must-Have-kostenlose öffentliche Datensatzwebsites Von Anfängern bis zu Experten: 10 Must-Have-kostenlose öffentliche Datensatzwebsites Sep 15, 2025 pm 03:51 PM

Für Anfänger in der Datenwissenschaft ist der Kern des Sprungs von "Unerfahrenheit" zum "Branchenexperten" eine kontinuierliche Praxis. Die Grundlage der Praxis sind die reichen und vielfältigen Datensätze. Glücklicherweise gibt es eine große Anzahl von Websites im Internet, die kostenlose öffentliche Datensätze anbieten, die wertvolle Ressourcen sind, um die Fähigkeiten zu verbessern und Ihre Fähigkeiten zu verbessern.

Was ist BIP? Warum sind sie für die Zukunft von Bitcoin so wichtig? Was ist BIP? Warum sind sie für die Zukunft von Bitcoin so wichtig? Sep 24, 2025 pm 01:51 PM

Inhaltsverzeichnis Was ist Bitcoin Improvement Vorschlag (BIP)? Warum ist BIP so wichtig? Wie funktioniert der historische BIP -Prozess für Bitcoin Improvement Vorschlag (BIP)? Was ist ein BIP -Typ -Signal und wie sendet ein Bergmann es? Taproot und Cons of Quick Trial of BIP -Schlussfolgerung - seit 2011 wurden Verbesserungen an Bitcoin durch ein System namens Bitcoin Improvement Vorschlag oder „BIP“ vorgenommen. Bitcoin Improvement Vorschlag (BIP) enthält Richtlinien dafür, wie sich Bitcoin im Allgemeinen entwickeln kann. Es gibt drei mögliche BIP -Arten, von denen zwei mit den technologischen Veränderungen in Bitcoin zusammenhängen. Jede BIP beginnt mit informellen Diskussionen zwischen Bitcoin -Entwicklern, die sich überall versammeln können, einschließlich TWI

Wie wähle ich einen Computer aus, der für die Big -Data -Analyse geeignet ist? Konfigurationshandbuch für High Performance Computing Wie wähle ich einen Computer aus, der für die Big -Data -Analyse geeignet ist? Konfigurationshandbuch für High Performance Computing Sep 15, 2025 pm 01:54 PM

Die Big-Data-Analyse muss sich auf Multi-Core-CPU, Speicher mit großer Kapazität und abgestufter Speicher konzentrieren. Multi-Core-Prozessoren wie Amdepyc oder Ryzenthreadripper werden unter Berücksichtigung der Anzahl der Kerne und einer Single-Core-Leistung bevorzugt. Der Speicher wird empfohlen, um mit 64 GB zu beginnen, und der ECC -Speicher wird bevorzugt, um die Datenintegrität zu gewährleisten. Der Speicher verwendet NVMESSD (System- und Heißdaten), Satassd (gemeinsame Daten) und HDD (Kaltdaten), um die Gesamtverarbeitungseffizienz zu verbessern.

Wie können Sie einen Kontextmanager mit dem @ContextManager -Dekorateur in Python erstellen? Wie können Sie einen Kontextmanager mit dem @ContextManager -Dekorateur in Python erstellen? Sep 20, 2025 am 04:50 AM

Importieren Sie@contextManagerfromContextLibanddefinaGeneratorFunctionThatyieldSexactlyonce, whercodeBeforyieldactsasenterandCodeafteryield (vorzugsweise infinal) actsas __exit __. 2.UsetheFunctionInaThstatement, wherheided ValuesieScessable

So schreiben Sie Automatisierungsskripte für tägliche Aufgaben in Python So schreiben Sie Automatisierungsskripte für tägliche Aufgaben in Python Sep 21, 2025 am 04:45 AM

Identifizieren Sie die RepetivetaSksworthautomating, SuchasorganizingFileSendingemails, FocusingontonTheSethatoccurfRequent und Takesineficanttime

See all articles