pequeñas ruedas cuda nn

Este repositorio facilita la creación de archivos de rueda de Python (.whl) desde el proyecto tiny-cuda-nn para agilizar el proceso de instalación en Google Colab y Kaggle. ¡Esto es para eludir el requisito de compilación de 20 minutos para tiny-cuda-nn en Google colab y Kaggle, cuando se realiza desde la fuente, para reducirlo a unos pocos segundos!

(Todos los créditos y licencias relevantes se atribuyen a Nvidia. Los materiales y las licencias de software del repositorio original de tiny-cuda-nn no están incluidos en este repositorio. Consulte el proyecto original para obtener detalles sobre la licencia).

El formato actual para los nombres de las ruedas incluye un sufijo de versión que indica la compatibilidad informática de la tarjeta gráfica correspondiente (es decir, la compatibilidad informática de 8.6 es .post86). Para simplificar, puede utilizar el siguiente código de Google Colab para el modelo de GPU correspondiente, pero Si desea ejecutarlo localmente en su máquina, siempre puede identificar la compatibilidad informática a través de esta página según su tarjeta gráfica https://developer.nvidia.com/cuda-gpus

También utiliza un sufijo de lanzamiento para cuda y torch .post{arch-type}{cuda-version}{torch-version} que indica la compatibilidad de torch y cuda.

Uso de Google Colab:

Para 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

Para GPU A100 y 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

Uso del cuaderno Kaggle:

Para 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

Para 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