Алтай
TensorFlow в Rust
Для чего нужна данная статья? : Написать собственный фреймворк, который: Зачем Вам это уметь? : Есть официальный Rust-крейт tensorflow, который предоставляет API для работы с моделями TensorFlow. Он позволяет: Пример загрузки модели и выполнения инференса: use tensorflow::{Graph, Session, SessionOptions, SessionRunArgs, Tensor}; fn main() -> tensorflow::Result<()> { let model_dir = "my_model"; // Путь к модели let mut graph = Graph::new(); let mut session = Session::new(&SessionOptions::new(), &graph)?; let input_tensor = Tensor::new(&[1])...
Ubuntu 22.04 + TensorFlow 2.16 + cudnn 8
Вот не ностальгия это, а "вьетнамские флешбеки". Почти весь день пришлось разбираться с убунтой и настраивать тензорфлоу. Но под конец все таки смог прикрутить то что нужно. Опущу пока проблемы самой Ubuntu (вот Вы знали что у последней убунты есть проблема если она стоит на SSD? И что попытка убунты загрузиться случается быстрее чем видеоадаптер загрузиться?) приведу пока команды для установки tensorflow + GPU 0) Всегда смотри таблицу совместимости. Для tensorflow это СВЕРХВАЖНО! 1) Драйвера. Принцип "Ставь последние стабильные" тут не работает, просто потому что версия CUDA тут 12...