Selesai: lumpuhkan gpu dalam notebook jupyter dalam aliran tensor

Kemaskini terakhir: 09/11/2023

GPU ialah alat yang berkuasa untuk mempercepatkan pengiraan pada set data yang besar. Walau bagaimanapun, melumpuhkan GPU kadangkala diperlukan untuk meningkatkan prestasi. Satu sebab biasa untuk melumpuhkan GPU ialah apabila menggunakan TensorFlow pada komputer riba atau komputer meja dengan sumber yang terhad. Melumpuhkan GPU juga boleh meningkatkan prestasi apabila menggunakan TensorFlow pada peranti mudah alih.

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

Baris pertama mengimport modul os. Baris kedua menetapkan pembolehubah persekitaran CUDA_VISIBLE_DEVICES kepada -1. Ini memberitahu TensorFlow untuk tidak menggunakan sebarang GPU.

Notebook Jupyter

Jupyter Notebook ialah aplikasi web yang membolehkan anda mencipta dan berkongsi dokumen dengan orang lain dalam persekitaran kolaboratif. Ia termasuk ciri seperti pemformatan teks kaya, pengeditan kod langsung dan penjanaan dokumen automatik.

Aliran Tensor

TensorFlow ialah perpustakaan untuk analisis data dan pembelajaran mesin. Ia membolehkan anda mencipta model dan algoritma tersuai untuk mempercepatkan latihan model tersebut. TensorFlow boleh digunakan dalam Python, yang menjadikannya mudah untuk bermula.

GPU

GPU dalam Python ialah alat berkuasa yang boleh digunakan untuk pelbagai tugas, termasuk pemprosesan grafik, pembelajaran mesin dan pengkomputeran saintifik. GPU ialah mikropemproses khusus yang direka untuk pemaparan grafik berprestasi tinggi dan pembelajaran mendalam. Python menyediakan pembungkus yang mudah untuk perpustakaan CUDA, yang memudahkan untuk mengakses sumber GPU.

Related posts: