keras/requirements-torch-cuda.txt

# Tensorflow cpu-only version (needed for testing).
tf-nightly-cpu==2.16.0.dev20240101  # Pin a working nightly until rc0.

# Torch with cuda support.
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.2
torchvision==0.16.2

# Jax cpu-only version (needed for testing).
jax[cpu]

-r requirements-common.txt
Add cuda installation instructions (#18566) Now that tf allows cuda via pip, we can finally provide some reasonable installation instructions. I am relying on tf-nightly (so that the Keras installation does not blow up tensorflow), but we could go back to a stable tf release. 2023-10-07 06:45:12 +00:00			`# Tensorflow cpu-only version (needed for testing).`
Bump the python group with 4 updates (#19008) Bumps the python group with 4 updates: [tf-nightly[and-cuda]](https://github.com/tensorflow/tensorflow), torch, torchvision and [tf-nightly-cpu](https://github.com/tensorflow/tensorflow). Updates `tf-nightly[and-cuda]` from 2.16.0.dev20231130 to 2.16.0.dev20240101 - [Release notes](https://github.com/tensorflow/tensorflow/releases) - [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md) - [Commits](https://github.com/tensorflow/tensorflow/commits) Updates `torch` from 2.1.1+cu118 to 2.1.2+cu118 Updates `torchvision` from 0.16.1+cu118 to 0.16.2+cu118 Updates `tf-nightly-cpu` from 2.16.0.dev20231130 to 2.16.0.dev20240101 - [Release notes](https://github.com/tensorflow/tensorflow/releases) - [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md) - [Commits](https://github.com/tensorflow/tensorflow/commits) --- updated-dependencies: - dependency-name: tf-nightly[and-cuda] dependency-type: direct:production update-type: version-update:semver-patch dependency-group: python - dependency-name: torch dependency-type: direct:production update-type: version-update:semver-patch dependency-group: python - dependency-name: torchvision dependency-type: direct:production update-type: version-update:semver-patch dependency-group: python - dependency-name: tf-nightly-cpu dependency-type: direct:production update-type: version-update:semver-patch dependency-group: python ... Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> 2024-01-01 14:03:33 +00:00			`tf-nightly-cpu==2.16.0.dev20240101 # Pin a working nightly until rc0.`
Add cuda installation instructions (#18566) Now that tf allows cuda via pip, we can finally provide some reasonable installation instructions. I am relying on tf-nightly (so that the Keras installation does not blow up tensorflow), but we could go back to a stable tf release. 2023-10-07 06:45:12 +00:00
			`# Torch with cuda support.`
Update Cuda Reqs to align Cuda versions (#19012) * Update index and extra url * Align keras requirments fo torch 2024-01-04 20:41:29 +00:00			`--extra-index-url https://download.pytorch.org/whl/cu121`
			`torch==2.1.2`
			`torchvision==0.16.2`
Add cuda installation instructions (#18566) Now that tf allows cuda via pip, we can finally provide some reasonable installation instructions. I am relying on tf-nightly (so that the Keras installation does not blow up tensorflow), but we could go back to a stable tf release. 2023-10-07 06:45:12 +00:00
			`# Jax cpu-only version (needed for testing).`
			`jax[cpu]`

			`-r requirements-common.txt`