wehub-resource-sync
25 lines
1.3 KiB
Plaintext
25 lines
1.3 KiB
Plaintext
# If you make changes below this line, please also make the corresponding changes to `dl-cpu-requirements.txt`!
|
|
|
|
tensorflow==2.15.1; python_version < '3.12' and (sys_platform != 'darwin' or platform_machine != 'arm64')
|
|
tensorflow-macos==2.15.1; python_version < '3.12' and sys_platform == 'darwin' and platform_machine == 'arm64'
|
|
tensorflow-probability==0.23.0; python_version < '3.12'
|
|
tensorflow-datasets; python_version < '3.12'
|
|
|
|
--extra-index-url https://download.pytorch.org/whl/cu128 # for GPU versions of torch, torchvision
|
|
--find-links https://data.pyg.org/whl/torch-2.7.0+cu128.html # for GPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv
|
|
# specifying explicit plus-notation below so pip overwrites the existing cpu verisons
|
|
torch==2.7.0+cu128
|
|
torchvision==0.22.0+cu128
|
|
torch-scatter==2.1.2+pt27cu128
|
|
torch-sparse==0.6.18+pt27cu128
|
|
torch-cluster==1.6.3+pt27cu128
|
|
torch-spline-conv==1.2.2+pt27cu128
|
|
|
|
cupy-cuda12x>=13.4.0; sys_platform != 'darwin'
|
|
cudf-cu12>=24.12.0; sys_platform != 'darwin'
|
|
nixl==1.2.0; sys_platform != 'darwin'
|
|
|
|
jax==0.4.33; python_version < '3.12' and sys_platform != 'darwin'
|
|
jaxlib==0.4.33; python_version < '3.12' and sys_platform != 'darwin'
|
|
jax-cuda12-plugin[cuda12]==0.4.33; python_version < '3.12' and sys_platform != 'darwin'
|