tiny cuda nn wheels

พื้นที่เก็บข้อมูลนี้อำนวยความสะดวกในการสร้างไฟล์ Python wheel (.whl) จากโปรเจ็กต์ Tiny-cuda-nn เพื่อปรับปรุงกระบวนการติดตั้งบน Google Colab และ Kaggle นี่เป็นการหลีกเลี่ยงข้อกำหนดในการสร้าง 20 นาทีสำหรับ Tiny-cuda-nn บน Google colab และ Kaggle เมื่อทำจากแหล่งที่มา เพื่อลดเหลือไม่กี่วินาที!

(เครดิตและใบอนุญาตที่เกี่ยวข้องทั้งหมดเป็นของ Nvidia วัสดุและใบอนุญาตซอฟต์แวร์จากพื้นที่เก็บข้อมูล Tiny-cuda-nn ดั้งเดิมไม่รวมอยู่ในพื้นที่เก็บข้อมูลนี้ โปรดดูรายละเอียดใบอนุญาตจากโครงการดั้งเดิม)

รูปแบบปัจจุบันสำหรับชื่อวงล้อประกอบด้วย postfix เวอร์ชันที่บ่งบอกถึงความเข้ากันได้ในการคำนวณของกราฟิกการ์ดที่เกี่ยวข้อง (เช่น ความเข้ากันได้ในการประมวลผลของ 8.6 คือ .post86) เพื่อความง่าย คุณสามารถใช้โค้ดด้านล่างสำหรับ Google Colab สำหรับรุ่น GPU ที่เกี่ยวข้องได้ แต่ หากคุณต้องการรันในเครื่องของคุณ คุณสามารถระบุความเข้ากันได้ของการคำนวณผ่านหน้านี้ตามกราฟิกการ์ดของคุณ https://developer.nvidia.com/cuda-gpus

นอกจากนี้ยังใช้ postfix สำหรับ cuda และ torch .post{arch-type}{cuda-version}{torch-version} ที่บ่งบอกถึงความเข้ากันได้ของ torch และ cuda

การใช้งาน Google Colab:

สำหรับ 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

สำหรับ A100 GPU และ 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

การใช้โน้ตบุ๊ค Kaggle:

สำหรับ 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

สำหรับ 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