From 8478cc8d5ab50702f1eed8ea7ed42e35141d48de Mon Sep 17 00:00:00 2001 From: Xiaodong Ye Date: Tue, 31 Jan 2023 12:09:07 +0800 Subject: [PATCH 1/2] Update pytorch docker image tag to 22.12 in README.md nvcr.io/nvidia/pytorch:21.02-py3 doesn't contain /usr/local/lib/python3.8/dist-packages/torch/lib/*, following the build instructions in README.md will fail with errors. --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index e39152a..6c2b778 100644 --- a/README.md +++ b/README.md @@ -56,13 +56,13 @@ $ apt-get install patchelf rapidjson-dev python3-dev ``` An appropriate PyTorch container from [NGC](https://ngc.nvidia.com) must be used. -For example, to build a backend that uses the 21.02 version of the PyTorch +For example, to build a backend that uses the 22.12 version of the PyTorch container from NGC: ``` $ mkdir build $ cd build -$ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_DOCKER_IMAGE="nvcr.io/nvidia/pytorch:21.02-py3" .. +$ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_DOCKER_IMAGE="nvcr.io/nvidia/pytorch:22.12-py3" .. $ make install ``` @@ -80,8 +80,8 @@ Currently, Triton requires that a specially patched version of PyTorch be used with the PyTorch backend. The full source for these PyTorch versions are available as Docker images from [NGC](https://ngc.nvidia.com). For example, the PyTorch version -compatible with the 21.02 release of Triton is available as -nvcr.io/nvidia/pytorch:21.02-py3. +compatible with the 22.12 release of Triton is available as +nvcr.io/nvidia/pytorch:22.12-py3. Copy over the LibTorch and Torchvision headers and libraries from the [PyTorch NGC container](https://ngc.nvidia.com/catalog/containers/nvidia:pytorch) From 24ae6ffe255401efbe9496df9f5dadba6910252d Mon Sep 17 00:00:00 2001 From: Xiaodong Ye Date: Fri, 10 Feb 2023 08:42:05 +0800 Subject: [PATCH 2/2] Update the copyright year in README.md Signed-off-by: Xiaodong Ye --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 6c2b778..c832ae5 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,5 @@