крошечные колеса cuda nn

Этот репозиторий упрощает создание файлов колеса Python (.whl) из проекта tiny-cuda-nn для оптимизации процесса установки в Google Colab и Kaggle. Это сделано для того, чтобы обойти требование 20 минут сборки tiny-cuda-nn в Google Colab и Kaggle, если оно выполняется из исходного кода, и сократить его до нескольких секунд!

(Все соответствующие авторские права и лицензии принадлежат Nvidia. Лицензии на материалы и программное обеспечение из исходного репозитория tiny-cuda-nn не включены в этот репозиторий. Подробную информацию о лицензировании см. в исходном проекте.)

Текущий формат названий колес включает постфикс версии, который обозначает вычислительную совместимость соответствующей видеокарты (т. е. вычислительная совместимость 8.6 – .post86). Для простоты вы можете использовать приведенный ниже код для Google Colab для соответствующей модели графического процессора, но если вы хотите запустить его локально на своем компьютере, вы всегда можете определить совместимость вычислений на этой странице на основе вашей видеокарты https://developer.nvidia.com/cuda-gpus

Он также использует постфикс выпуска для cuda и torch .post{arch-type}{cuda-version}{torch-version}, который означает совместимость torch и cuda.

Использование Google Colab:

Для графического процессора 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

Для графического процессора A100 и графического процессора 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

Использование ноутбука Kaggle:

Для графического процессора 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

Для графического процессора 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