roda cuda nn kecil

Repositori ini memfasilitasi pembuatan file roda Python (.whl) dari proyek tiny-cuda-nn untuk menyederhanakan proses instalasi di Google Colab dan Kaggle. Hal ini untuk menghindari persyaratan pembuatan 20 menit untuk tiny-cuda-nn di Google colab dan Kaggle, jika dilakukan dari sumbernya, untuk menguranginya menjadi beberapa detik!

(Semua kredit dan lisensi yang relevan diberikan kepada Nvidia. Materi dan lisensi perangkat lunak dari repositori tiny-cuda-nn asli tidak disertakan dalam repositori ini. Silakan merujuk ke proyek asli untuk detail lisensi.)

Format nama roda saat ini menyertakan postfix rilis yang menandakan kompatibilitas komputasi kartu grafis yang relevan (yaitu kompatibilitas komputasi 8.6 adalah .post86), untuk kesederhanaan Anda dapat menggunakan kode di bawah ini untuk Google Colab untuk model GPU yang relevan, tetapi jika Anda ingin menjalankannya secara lokal di mesin Anda, Anda selalu dapat mengidentifikasi kompatibilitas komputasi melalui halaman ini berdasarkan kartu grafis Anda https://developer.nvidia.com/cuda-gpus

Ia juga menggunakan postfix rilis untuk cuda dan torch .post{arch-type}{cuda-version}{torch-version} yang menandakan kompatibilitas torch dan cuda.

Penggunaan Google Colab:

Untuk GPU T4

!curl -L "https://github.com/OutofAi/tiny-cuda-nn-wheels/releases/download/1.7.2/tinycudann-1.7.post75240121-cp310-cp310-linux_x86_64.whl" -o tinycudann-1.7.post75240121-cp310-cp310-linux_x86_64.whl
!pip install tinycudann-1.7.post75240121-cp310-cp310-linux_x86_64.whl --force-reinstall
import tinycudann as tcnn

Untuk GPU A100 dan GPU L4

!curl -L "https://github.com/OutofAi/tiny-cuda-nn-wheels/releases/download/1.7.2/tinycudann-1.7.post89240121-cp310-cp310-linux_x86_64.whl" -o tinycudann-1.7.post89240121-cp310-cp310-linux_x86_64.whl
!pip install tinycudann-1.7.post89240121-cp310-cp310-linux_x86_64.whl --force-reinstall
import tinycudann as tcnn

Penggunaan Buku Catatan Kaggle:

Untuk GPU T4

!curl -L "https://github.com/OutofAi/tiny-cuda-nn-wheels/releases/download/Kaggle-T4/tinycudann-1.7-cp310-cp310-linux_x86_64.whl" -o tinycudann-1.7-cp310-cp310-linux_x86_64.whl
!python -m pip install tinycudann-1.7-cp310-cp310-linux_x86_64.whl --force-reinstall --no-cache-dir
import tinycudann as tcnn

Untuk GPU P100

!curl -L "https://github.com/OutofAi/tiny-cuda-nn-wheels/releases/download/Kaggle-P100/tinycudann-1.7-cp310-cp310-linux_x86_64.whl" -o tinycudann-1.7-cp310-cp310-linux_x86_64.whl
!python -m pip install tinycudann-1.7-cp310-cp310-linux_x86_64.whl --force-reinstall --no-cache-dir
import tinycudann as tcnn