tch-rs
tch-rs copied to clipboard
LaurentMazare
Rust bindings for the C++ api of PyTorch.
I don't know much about this question, but I also met this question when I want to use the same tensor in multiple threads. After a lot of trying, I...
I added a simple example that shows how to run our custom-trained PyTorch models on a webcam with OpenCV and tch-rs. The example runs on GPU. * EDIT1: Current version...
I was using a machine with eight graphics cards to train a model. However, I didn't know how to use 2 or more GPU at the same time to train...
In the case of using `tch` as a sub-dependency of [other crates](https://crates.io/crates/tch/reverse_dependencies), disabling the default feature of downloading `libtorch` when depending on any of them becomes impossible. According to [this...
The work is an attempt to support third-party C++/CUDA extensions in compile time. The current `torch-sys/src/build.rs` design runs all linking jobs for `torch-sys` crate only, making it difficult to link...
Do you know of any examples of Python modules written in Rust using tch? I'm interested in implementing a custom RNN cell in Rust using tch and exposing it to...
Hi, i want to know the process of making the yolo model on the example file, i always getting error when i try to load my own model (ofc after...
Hello. I'm trying to compile a project that uses your library as a dependency. I'm currently on Linux and, as per instructions in the readme about manual installations, I added...
I do a lot of NLP, and after tokenizing and converting to indexes, my tokens are of type usize. Is it possible to impl Element for usize so a type...
Metadata
Owner
Metadata
Rust bindings for the C++ api of PyTorch.