We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 33c0bc4 commit d600509Copy full SHA for d600509
README.md
@@ -56,13 +56,13 @@ $ apt-get install patchelf rapidjson-dev python3-dev
56
```
57
58
An appropriate PyTorch container from [NGC](https://ngc.nvidia.com) must be used.
59
-For example, to build a backend that uses the 22.12 version of the PyTorch
+For example, to build a backend that uses the 23.04 version of the PyTorch
60
container from NGC:
61
62
63
$ mkdir build
64
$ cd build
65
-$ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_DOCKER_IMAGE="nvcr.io/nvidia/pytorch:22.12-py3" ..
+$ cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install -DTRITON_PYTORCH_DOCKER_IMAGE="nvcr.io/nvidia/pytorch:23.04-py3" ..
66
$ make install
67
68
0 commit comments