Skip to content

Commit f40cbc1

Browse files
committed
Better instructions
1 parent 4fd6cb7 commit f40cbc1

File tree

1 file changed

+25
-19
lines changed

1 file changed

+25
-19
lines changed

README.md

Lines changed: 25 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -146,12 +146,7 @@ First, make sure you have a GPU that has NVDEC hardware that can decode the
146146
format 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 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.
155150
TorchCodec with CUDA should work with FFmpeg versions in [4, 7].
156151

157152
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
178173
ffmpeg -hwaccel cuda -hwaccel_output_format cuda -i test/resources/nasa_13013.mp4 -f null -
179174
```
180175

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+
181184
3. Install TorchCodec
182185

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:
185188

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+
```
191194

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.
194197

195-
- **On Windows** (experimental support), you'll need to rely on `conda`:
198+
#### Windows
196199

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+
```
200206

201207
## Benchmark Results
202208

0 commit comments

Comments
 (0)