{"repo":"sugarme/gotch","free":true,"listed":false,"github":"https://github.com/sugarme/gotch","clone":"git clone https://github.com/sugarme/gotch.git","description":"Go binding for Pytorch C++ API (libtorch)","language":"Go","stars":653,"topics":["pytorch","go","golang","deep-learning","framework","go-binding","pytorch-go-binding"],"license":"Apache-2.0","category":"machine-learning","readme_excerpt":"Gotch Overview gotch creates a thin wrapper to Pytorch C++ APIs (Libtorch) to make use of its already optimized C++ tensor APIs (3039) and dynamic graph computation with CUDA support and provides idiomatic Go APIs for developing and implementing Deep Learning in Go. Some features are - [x] Comprehensive Pytorch tensor APIs (2525) - [x] Fully featured Pytorch dynamic graph computation - [x] JIT interface to run model trained/saved using PyTorch Python API - [x] Load pretrained Pytorch models and run inference - [x] Pure Go APIs to build and train neural network models with both CPU and GPU support - [x] Most recent image models - [ ] NLP Language models - Transformer in separate package built with gotch and pure Go Tokenizer. gotch is in active development mode and may have API breaking changes. Feel free to pull request, report issues or discuss any concerns. All contributions are welcome. gotch current version is v0.9.1 Dependencies - Libtorch C++ v2.1.0 library of Pytorch - Clang-17/Clang++-17 compilers Installation - Default CUDA version is 11.8 if CUDA is available otherwise using CPU version. - Default Pytorch C++ API version is 2.1.0 NOTE : libtorch will be installed at /usr/local/lib CPU Step 1: Setup libtorch (skip this step if a valid libtorch already installed in your machine!) Update Environment : in Debian/Ubuntu, add/update the following lines to .bashrc file Step 2: Setup gotch GPU NOTE : make sure your machine has working CUDA. - Check version: nvidia-smi - Ins","default_branch":null,"files":null,"tree":[],"storefront":"/r/sugarme","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/sugarme/gotch/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}