@@ -146,14 +146,12 @@ First, make sure you have a GPU that has NVDEC hardware that can decode the
146146format you want. Refer to Nvidia's GPU support matrix for more details
147147[ here] ( https://developer.nvidia.com/video-encode-and-decode-gpu-support-matrix-new ) .
148148
149- 1 . Install CUDA Toolkit. Pytorch and TorchCodec supports CUDA Toolkit
150- versions 11.8, 12.1 or 12.4. In particular TorchCodec depends on
151- CUDA libraries libnpp and libnvrtc (which are part of CUDA Toolkit).
149+ 1 . Install Pytorch corresponding to your CUDA Toolkit using the
150+ [ official instructions] ( https://pytorch.org/get-started/locally/ ) . You'll
151+ need the ` libnpp ` and ` libnvrtc ` CUDA libraries, which are usually part of
152+ the CUDA Toolkit.
152153
153- 2 . Install Pytorch that corresponds to your CUDA Toolkit version using the
154- [ official instructions] ( https://pytorch.org/get-started/locally/ ) .
155-
156- 3 . Install or compile FFmpeg with NVDEC support.
154+ 2 . Install or compile FFmpeg with NVDEC support.
157155 TorchCodec with CUDA should work with FFmpeg versions in [ 5, 7] .
158156
159157 If FFmpeg is not already installed, or you need a more recent version, an
@@ -184,13 +182,13 @@ format you want. Refer to Nvidia's GPU support matrix for more details
184182 ffmpeg -hwaccel cuda -hwaccel_output_format cuda -i test/resources/nasa_13013.mp4 -f null -
185183 ```
186184
187- 4 . Install TorchCodec by passing in an ` --index-url ` parameter that corresponds
185+ 3 . Install TorchCodec by passing in an ` --index-url ` parameter that corresponds
188186 to your CUDA Toolkit version, example:
189187
190188 ``` bash
191- # This corresponds to CUDA Toolkit version 12.4 . It should be the same one
189+ # This corresponds to CUDA Toolkit version 12.6 . It should be the same one
192190 # you used when you installed PyTorch (If you installed PyTorch with pip).
193- pip install torchcodec --index-url=https://download.pytorch.org/whl/cu124
191+ pip install torchcodec --index-url=https://download.pytorch.org/whl/cu126
194192 ```
195193
196194 Note that without passing in the ` --index-url ` parameter, ` pip ` installs
0 commit comments