pequenas rodas cuda nn

Este repositório facilita a criação de arquivos roda Python (.whl) do projeto tiny-cuda-nn para agilizar o processo de instalação no Google Colab e Kaggle. Isso é para contornar o requisito de compilação de 20 minutos para tiny-cuda-nn no Google colab e Kaggle, quando feito a partir da fonte, para reduzi-lo para alguns segundos!

(Todos os créditos e licenças relevantes são atribuídos à Nvidia. Os materiais e licenças de software do repositório tiny-cuda-nn original não estão incluídos neste repositório. Consulte o projeto original para obter detalhes de licenciamento.)

O formato atual para os nomes das rodas inclui um postfix de lançamento que significa a compatibilidade computacional da placa gráfica relevante (ou seja, a compatibilidade computacional de 8.6 é .post86). Para simplificar, você pode usar o código abaixo para Google Colab para o modelo de GPU relevante, mas se quiser executá-lo localmente em sua máquina você sempre pode identificar a compatibilidade computacional através desta página com base em sua placa gráfica https://developer.nvidia.com/cuda-gpus

Ele também usa um postfix de lançamento para cuda e torch .post{arch-type}{cuda-version}{torch-version} que significa a compatibilidade do torch e do cuda.

Uso do 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 e 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 do notebook 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