Bagikan melalui

Mengaktifkan NVIDIA CUDA di WSL

Pembaruan Windows 11 dan yang lebih baru dari Windows 10 mendukung menjalankan alat ML, pustaka, dan kerangka kerja populer yang ada yang menggunakan NVIDIA CUDA untuk akselerasi perangkat keras GPU di dalam instans Subsistem Windows untuk Linux (WSL). Ini termasuk PyTorch dan TensorFlow serta semua dukungan Docker dan NVIDIA Container Toolkit yang tersedia di lingkungan Linux asli.

Menginstal Windows 11 atau Windows 10, versi 21H2

Untuk menggunakan fitur-fitur ini, Anda dapat mengunduh dan menginstal Windows 11 atau Windows 10, versi 21H2.

Menginstal driver GPU

Unduh dan instal driver NVIDIA yang mendukung CUDA untuk WSL agar dapat digunakan dengan alur kerja CUDA ML Anda yang ada. Untuk informasi selengkapnya tentang driver mana yang akan diinstal, lihat:

Menginstal WSL

Setelah menginstal driver di atas, pastikan Anda mengaktifkan WSL dan menginstal distribusi berbasis glibc, seperti Ubuntu atau Debian. Pastikan Anda memiliki kernel terbaru dengan memilih Periksa pembaruan di bagian Pembaruan Windows di aplikasi Pengaturan.

Nota

Pastikan Anda mengaktifkan Terima pembaruan untuk produk Microsoft lainnya . Anda dapat menemukannya di opsi Tingkat Lanjut di dalam bagian Windows Update di aplikasi Pengaturan.

Untuk fitur-fitur ini, Anda memerlukan versi kernel 5.10.43.3 atau yang lebih tinggi. Anda dapat memeriksa nomor versi dengan menjalankan perintah berikut di PowerShell.

wsl cat /proc/version

Mulai menggunakan NVIDIA CUDA

Sekarang ikuti instruksi dalam NVIDIA CUDA di Panduan Pengguna WSL dan Anda dapat mulai menggunakan alur kerja Linux yang sedang eksisi melalui NVIDIA Docker, atau dengan menginstal PyTorch atau TensorFlow di dalam WSL.

Bagikan umpan balik tentang dukungan NVIDIA melalui forum Komunitas mereka untuk CUDA di WSL.