Skip to content

Commit

Permalink
Patch pytorch-cpu/gpu for 2.4.1 increased compatibility
Browse files Browse the repository at this point in the history
  • Loading branch information
hmaarrfk committed Feb 1, 2025
1 parent f43ade7 commit 6fbe6c8
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions recipe/patch_yaml/pytorch.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -160,3 +160,29 @@ if:

then:
- add_constrains: pybind11 <0.0.0a0


---
# Pytorch 2.4.1 did not upload a package for pytorch-gpu and pytorch-cpu that
# compatible with all versions of python
# Only the linux builds use the "megabuild" that cause the issue
# The OSX builds were "safe"
# https://github.com/conda-forge/pytorch-cpu-feedstock/issues/338
# https://github.com/conda-forge/pytorch-cpu-feedstock/pull/341

if:
name: pytorch-gpu
version: 2.4.1
subdir_in: linux-64
build_number_in: [0, 1, 2, 3, 4, 5, 100, 101, 102, 103, 104, 105, 200, 201, 202, 203, 204, 205, 300, 301, 302, 303, 304, 305]

then:
- replace_depends:
old: pytorch ${version} cuda118_*${build_number}
new: pytorch ${version} cuda118_*${build_number}
- replace_depends:
old: pytorch ${version} cuda120_*${build_number}
new: pytorch ${version} cuda120_*${build_number}
- replace_depends:
old: pytorch ${version} cpu_
new: pytorch ${version} cpu_*${build_number}

0 comments on commit 6fbe6c8

Please sign in to comment.