Skip to content

Commit

Permalink
update GPU description
Browse files Browse the repository at this point in the history
  • Loading branch information
KJlaccHoeUM9l authored and vvchernov committed Feb 10, 2022
1 parent bdf3291 commit 3845cf4
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion docs/TVM_EP.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,9 @@ apt-get install -y python3 python3-dev python3-pip python3-setuptools gcc libtin
pip3 install numpy decorator attrs
```

Also, the current implementation has `GPU` support for TVM EP. For now, you can use an `NVidia GPU` with CUDA Toolkit support. To do this, make sure you have installed the NVidia driver and CUDA Toolkit. More detailed instructions can be found on the [official page](https://developer.nvidia.com/cuda-toolkit).
Also, the current implementation has `NVidia GPU` support for TVM EP. For now, you can use only `NVidia GPU` with CUDA Toolkit support.
To do this, make sure you have installed the NVidia driver and CUDA Toolkit.
More detailed instructions can be found on the [official page](https://developer.nvidia.com/cuda-toolkit).

Clone this repo.
In order to build ONNXRT you will need to have CMake 3.18 or higher. In Ubuntu 20.04 you can use the following commands to install the latest version of CMake:
Expand Down

0 comments on commit 3845cf4

Please sign in to comment.