عجلات 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 لنموذج GPU ذي الصلة، ولكن إذا كنت تريد تشغيله محليًا على جهازك، فيمكنك دائمًا تحديد توافق الحساب من خلال هذه الصفحة بناءً على بطاقة الرسومات الخاصة بك https://developer.nvidia.com/cuda-gpus

ويستخدم أيضًا إصدارًا لاحقاً لـ cuda وtorch .post{arch-type}{cuda-version}{torch-version} الذي يشير إلى توافق الشعلة وcuda.

استخدام جوجل كولاب:

لوحدة معالجة الرسومات 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