sputnik

สปุตนิก

Sputnik เป็นห้องสมุดของเมล็ดพีชคณิตเชิงเส้นเบาบางและยูทิลิตี้สำหรับการเรียนรู้อย่างลึกซึ้ง

สร้าง

Sputnik ใช้ระบบ CMake Build สปุตนิกขึ้นอยู่กับชุดเครื่องมือ CUDA (v10.1+) และรองรับ SM70+ การพึ่งพาเพิ่มเติมสำหรับไลบรารีคือ Google/glog ในการสร้างไลบรารีให้ป้อนไดเรกทอรีโครงการและเรียกใช้คำสั่งต่อไปนี้:

mkdir build && cd build

cmake .. -DCMAKE_BUILD_TYPE=Release

make -j12

ห้องทดสอบและเบนช์มาร์กเพิ่มเติมขึ้นอยู่กับ Abseil/Abseil-CPP, Google/Googltest และ Google/Benchmark การพึ่งพาเหล่านี้รวมเป็น submodules ใน third_party ในการสร้างชุดทดสอบและ/หรือ Benchmark Suite ให้ตั้งค่า -DBUILD_TEST=ON และ/หรือ -DBUILD_BENCHMARK=ON ในคำสั่ง cmake ของคุณ

cmake .. -DCMAKE_BUILD_TYPE=Release -DBUILD_TEST=ON -DBUILD_BENCHMARK=ON -DCUDA_ARCHS="70;75"

นักเทียบท่า

Sputnik จัดเตรียม Dockerfile ที่สร้างสภาพแวดล้อมที่เหมาะสมด้วยการพึ่งพาทั้งหมด โปรดทราบว่าจะต้องติดตั้ง Nvidia-Docker เพื่อทำงานบน GPU ในการสร้างภาพเรียกใช้คำสั่งต่อไปนี้:

docker build . -t sputnik-dev

ในการเปิดตัวคอนเทนเนอร์ด้วยแหล่งกำเนิดสปุตนิกที่ติดตั้งอยู่ใต้ /mount

sudo docker run --runtime=nvidia -v ~/:/mount/ -it sputnik-dev:latest

การอ้างอิง

หากคุณใช้ประโยชน์จากห้องสมุดนี้โปรดอ้างอิง:

 @inproceedings{sgk_sc2020,
  author    = {Trevor Gale and Matei Zaharia and Cliff Young and Erich Elsen},
  title     = {Sparse {GPU} Kernels for Deep Learning},
  booktitle = {Proceedings of the International Conference for High Performance Computing, Networking, Storage and Analysis, {SC} 2020},
  year      = {2020},
}

โมเดลที่กระจัดกระจายและชุดข้อมูลของเมทริกซ์เบาบางจากเครือข่ายประสาทลึกจากกระดาษด้านบนสามารถพบได้ที่นี่

คำเตือน

นี่ไม่ใช่ผลิตภัณฑ์ของ Google อย่างเป็นทางการ