From 454db975d89d720e6f9402832495bacccae3d317 Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Fri, 23 Aug 2024 15:00:20 +0900 Subject: [PATCH] update readme --- rust/tvm/examples/resnet/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rust/tvm/examples/resnet/README.md b/rust/tvm/examples/resnet/README.md index d6e32f7fa768e..ad76ac0048a02 100644 --- a/rust/tvm/examples/resnet/README.md +++ b/rust/tvm/examples/resnet/README.md @@ -21,7 +21,7 @@ This end-to-end example shows how to: * build `Resnet 18` with `tvm` from Python * use the provided Rust frontend API to test for an input image -To run the example with pretrained resnet weights, first `tvm` and `mxnet` must be installed for the python build. To install mxnet for cpu, run `pip install mxnet` +To run the example with pretrained resnet weights, first `tvm` and `torchvision` must be installed for the python build. To install torchvision for cpu, run `pip install torch torchvision` and to install `tvm` with `llvm` follow the [TVM installation guide](https://tvm.apache.org/docs/install/index.html). * **Build the example**: `cargo build