@@ -146,12 +146,7 @@ First, make sure you have a GPU that has NVDEC hardware that can decode the
146
146
format you want. Refer to Nvidia's GPU support matrix for more details
147
147
[ here] ( https://developer.nvidia.com/video-encode-and-decode-gpu-support-matrix-new ) .
148
148
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.
153
-
154
- 2 . Install or compile FFmpeg with NVDEC support.
149
+ 1 . Install FFmpeg with NVDEC support.
155
150
TorchCodec with CUDA should work with FFmpeg versions in [ 4, 7] .
156
151
157
152
If FFmpeg is not already installed, or you need a more recent version, an
@@ -178,25 +173,36 @@ format you want. Refer to Nvidia's GPU support matrix for more details
178
173
ffmpeg -hwaccel cuda -hwaccel_output_format cuda -i test/resources/nasa_13013.mp4 -f null -
179
174
```
180
175
176
+ #### Linux
177
+
178
+ 2 . Install Pytorch corresponding to your CUDA Toolkit using the
179
+ [ official instructions] ( https://pytorch.org/get-started/locally/ ) . You'll
180
+ need the ` libnpp ` and ` libnvrtc ` CUDA libraries, which are usually part of
181
+ the CUDA Toolkit.
182
+
183
+
181
184
3 . Install TorchCodec
182
185
183
- - ** On Linux ** , pass in an ` --index-url ` parameter that corresponds
184
- to your CUDA Toolkit version, for example:
186
+ Pass in an ` --index-url ` parameter that corresponds to your CUDA Toolkit
187
+ version, for example:
185
188
186
- ``` bash
187
- # This corresponds to CUDA Toolkit version 12.6. It should be the same one
188
- # you used when you installed PyTorch (If you installed PyTorch with pip).
189
- pip install torchcodec --index-url=https://download.pytorch.org/whl/cu126
190
- ```
189
+ ``` bash
190
+ # This corresponds to CUDA Toolkit version 12.6. It should be the same one
191
+ # you used when you installed PyTorch (If you installed PyTorch with pip).
192
+ pip install torchcodec --index-url=https://download.pytorch.org/whl/cu126
193
+ ```
191
194
192
- Note that without passing in the ` --index-url` parameter, ` pip` installs
193
- the CPU-only version of TorchCodec.
195
+ Note that without passing in the ` --index-url ` parameter, ` pip ` installs
196
+ the CPU-only version of TorchCodec.
194
197
195
- - ** On Windows** (experimental support), you ' ll need to rely on `conda`:
198
+ #### Windows
196
199
197
- ```bash
198
- conda install torchcodec -c conda-forge
199
- ```
200
+ On Windows (experimental support), you'll need to rely on ` conda ` to install
201
+ both pytorch and TorchCodec:
202
+
203
+ ``` bash
204
+ conda install -c conda-forge torchcodec=* =* cuda*
205
+ ```
200
206
201
207
## Benchmark Results
202
208
0 commit comments