1
1
<!--
2
- # Copyright 2020-2022 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
+ # Copyright 2020-2023 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3
3
#
4
4
# Redistribution and use in source and binary forms, with or without
5
5
# modification, are permitted provided that the following conditions
@@ -56,13 +56,13 @@ $ apt-get install patchelf rapidjson-dev python3-dev
56
56
```
57
57
58
58
An 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
60
60
container from NGC:
61
61
62
62
```
63
63
$ mkdir build
64
64
$ 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" ..
66
66
$ make install
67
67
```
68
68
@@ -80,8 +80,8 @@ Currently, Triton requires that a specially patched version of
80
80
PyTorch be used with the PyTorch backend. The full source for
81
81
these PyTorch versions are available as Docker images from
82
82
[ 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.
85
85
86
86
Copy over the LibTorch and Torchvision headers and libraries from the
87
87
[ PyTorch NGC container] ( https://ngc.nvidia.com/catalog/containers/nvidia:pytorch )
0 commit comments