tiny cuda nn wheels

Dieses Repository erleichtert die Erstellung von Python-Wheel-Dateien (.whl) aus dem Projekt tiny-cuda-nn, um den Installationsprozess auf Google Colab und Kaggle zu optimieren. Dies dient dazu, die 20-minütige Build-Anforderung für tiny-cuda-nn auf Google Colab und Kaggle zu umgehen, um sie auf wenige Sekunden zu reduzieren, wenn die Erstellung aus der Quelle erfolgt!

(Alle relevanten Credits und Lizenzen werden Nvidia zugeschrieben. Die Materialien und Softwarelizenzen aus dem ursprünglichen Tiny-Cuda-NN-Repository sind nicht in diesem Repository enthalten. Einzelheiten zur Lizenzierung finden Sie im Originalprojekt.)

Das aktuelle Format für die Radnamen enthält ein Release-Postfix, das die Rechenkompatibilität der relevanten Grafikkarte angibt (d. h. die Rechenkompatibilität von 8.6 ist .post86). Der Einfachheit halber können Sie den folgenden Code für Google Colab für das entsprechende GPU-Modell verwenden, aber Wenn Sie es lokal auf Ihrem Computer ausführen möchten, können Sie die Compute-Kompatibilität immer auf dieser Seite basierend auf Ihrer Grafikkarte ermitteln: https://developer.nvidia.com/cuda-gpus

Es verwendet außerdem einen Release-Postfix für Cuda und Torch .post{arch-type}{cuda-version}{torch-version}, der die Kompatibilität von Torch und Cuda angibt.

Google Colab-Nutzung:

Für T4-GPU

!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

Für A100-GPU und L4-GPU

!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

Verwendung des Kaggle-Notizbuchs:

Für T4-GPU

!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

Für P100-GPU

!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