sputnik

Sputnik

Sputnik adalah perpustakaan kernel aljabar linier yang jarang dan utilitas untuk pembelajaran yang mendalam.

Membangun

Sputnik menggunakan sistem build CMake. Sputnik tergantung pada CUDA Toolkit (V10.1+) dan mendukung SM70+. Satu -satunya ketergantungan tambahan untuk perpustakaan adalah Google/Glog. Untuk membangun perpustakaan, masukkan direktori proyek dan jalankan perintah berikut:

mkdir build && cd build

cmake .. -DCMAKE_BUILD_TYPE=Release

make -j12

Suite tes dan patokan juga bergantung pada Abseil/Abseil-CPP, Google/GooglTest, dan Google/Benchmark. Ketergantungan ini dimasukkan sebagai submodul di Third_party. Untuk membangun suite tes dan/atau rangkaian benchmark, atur -DBUILD_TEST=ON dan/atau -DBUILD_BENCHMARK=ON in cmake command Anda.

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

Buruh pelabuhan

Sputnik menyediakan dockerfile yang membangun lingkungan yang tepat dengan semua ketergantungan. Perhatikan bahwa nvidia-docker harus diinstal untuk dijalankan pada GPU. Untuk membangun gambar, jalankan perintah berikut:

docker build . -t sputnik-dev

Untuk meluncurkan wadah dengan sumber Sputnik yang dipasang di bawah /mount (dengan asumsi Anda bekerja di luar $ rumah), jalankan yang berikut:

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

Kutipan

Jika Anda menggunakan perpustakaan ini, silakan kutip:

 @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},
}

Model yang jarang dan dataset matriks jarang dari jaringan saraf dalam dari kertas di atas dapat ditemukan di sini.

Penafian

Ini bukan produk Google resmi.