11<!--
2- # Copyright 2020-2022 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+ # Copyright 2020-2023 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33#
44# Redistribution and use in source and binary forms, with or without
55# modification, are permitted provided that the following conditions
@@ -56,13 +56,13 @@ $ apt-get install patchelf rapidjson-dev python3-dev
5656```
5757
5858An appropriate PyTorch container from [ NGC] ( https://ngc.nvidia.com ) must be used.
59- For example, to build a backend that uses the 21.02 version of the PyTorch
59+ For example, to build a backend that uses the 22.12 version of the PyTorch
6060container from NGC:
6161
6262```
6363$ mkdir build
6464$ cd build
65- $ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_DOCKER_IMAGE="nvcr.io/nvidia/pytorch:21.02 -py3" ..
65+ $ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_DOCKER_IMAGE="nvcr.io/nvidia/pytorch:22.12 -py3" ..
6666$ make install
6767```
6868
@@ -80,8 +80,8 @@ Currently, Triton requires that a specially patched version of
8080PyTorch be used with the PyTorch backend. The full source for
8181these PyTorch versions are available as Docker images from
8282[ NGC] ( https://ngc.nvidia.com ) . For example, the PyTorch version
83- compatible with the 21.02 release of Triton is available as
84- nvcr.io/nvidia/pytorch:21.02 -py3.
83+ compatible with the 22.12 release of Triton is available as
84+ nvcr.io/nvidia/pytorch:22.12 -py3.
8585
8686Copy over the LibTorch and Torchvision headers and libraries from the
8787[ PyTorch NGC container] ( https://ngc.nvidia.com/catalog/containers/nvidia:pytorch )
0 commit comments