-
Notifications
You must be signed in to change notification settings - Fork 64
Add install instructions for Windows GPU #888
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 3 commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change | ||||||||
---|---|---|---|---|---|---|---|---|---|---|
|
@@ -131,6 +131,7 @@ The following table indicates the compatibility between versions of | |||||||||
| `torchcodec` | `torch` | Python | | ||||||||||
| ------------------ | ------------------ | ------------------- | | ||||||||||
| `main` / `nightly` | `main` / `nightly` | `>=3.10`, `<=3.13` | | ||||||||||
| `0.7` | `2.8` | `>=3.9`, `<=3.13` | | ||||||||||
| `0.6` | `2.8` | `>=3.9`, `<=3.13` | | ||||||||||
| `0.5` | `2.7` | `>=3.9`, `<=3.13` | | ||||||||||
| `0.4` | `2.7` | `>=3.9`, `<=3.13` | | ||||||||||
|
@@ -162,10 +163,6 @@ format you want. Refer to Nvidia's GPU support matrix for more details | |||||||||
conda install "ffmpeg<8" -c conda-forge | ||||||||||
``` | ||||||||||
|
||||||||||
If you are building FFmpeg from source you can follow Nvidia's guide to | ||||||||||
configuring and installing FFmpeg with NVDEC support | ||||||||||
[here](https://docs.nvidia.com/video-technologies/video-codec-sdk/12.0/ffmpeg-with-nvidia-gpu/index.html). | ||||||||||
|
||||||||||
After installing FFmpeg make sure it has NVDEC support when you list the supported | ||||||||||
decoders: | ||||||||||
|
||||||||||
|
@@ -181,17 +178,25 @@ format you want. Refer to Nvidia's GPU support matrix for more details | |||||||||
ffmpeg -hwaccel cuda -hwaccel_output_format cuda -i test/resources/nasa_13013.mp4 -f null - | ||||||||||
``` | ||||||||||
|
||||||||||
3. Install TorchCodec by passing in an `--index-url` parameter that corresponds | ||||||||||
to your CUDA Toolkit version, example: | ||||||||||
3. Install TorchCodec | ||||||||||
|
||||||||||
```bash | ||||||||||
# This corresponds to CUDA Toolkit version 12.6. It should be the same one | ||||||||||
# you used when you installed PyTorch (If you installed PyTorch with pip). | ||||||||||
pip install torchcodec --index-url=https://download.pytorch.org/whl/cu126 | ||||||||||
``` | ||||||||||
- **On Linux**, pass in an `--index-url` parameter that corresponds | ||||||||||
to your CUDA Toolkit version, for example: | ||||||||||
|
||||||||||
```bash | ||||||||||
# This corresponds to CUDA Toolkit version 12.6. It should be the same one | ||||||||||
# you used when you installed PyTorch (If you installed PyTorch with pip). | ||||||||||
pip install torchcodec --index-url=https://download.pytorch.org/whl/cu126 | ||||||||||
``` | ||||||||||
|
||||||||||
Note that without passing in the `--index-url` parameter, `pip` installs | ||||||||||
the CPU-only version of TorchCodec. | ||||||||||
|
||||||||||
- **On Windows** (experimental support), you'll need to rely on `conda`: | ||||||||||
|
||||||||||
Note that without passing in the `--index-url` parameter, `pip` installs | ||||||||||
the CPU-only version of TorchCodec. | ||||||||||
```bash | ||||||||||
conda install torchcodec -c conda-forge | ||||||||||
|
conda install torchcodec -c conda-forge | |
conda install torchcodec -c conda-forge |
this should indeed install torchcodec with cuda
support on a machine with cuda support, but it may silently install the cpu version on a machine without cuda support. If you want to make sure that the cuda version is installed, and an error is printed in case it is not possible to install, you need to:
conda install torchcodec -c conda-forge | |
conda install torchcodec=*=*cuda* -c conda-forge |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Drive-by, I don't want to bother readers about installing FFmpeg from source, they shouldn't need to and it shouldn't be in the "entry point" install instructions anyway.