Skip to content
Snippets Groups Projects

Make pytorch packages to depend on the cpuonly package

Merged Amir MOHAMMADI requested to merge pytorch into master
1 file
+ 5
0
Compare changes
  • Side-by-side
  • Inline
@@ -110,6 +110,11 @@ def _gen_new_index(repodata, packages_key):
@@ -110,6 +110,11 @@ def _gen_new_index(repodata, packages_key):
if parse_version(record["version"]) <= parse_version("4.1.0"):
if parse_version(record["version"]) <= parse_version("4.1.0"):
record["depends"].append("numpy <1.18")
record["depends"].append("numpy <1.18")
 
# somehow conda cannot resolve pytorch cpu without the cpuonly package
 
# we only ship cpu-only pytorch packages
 
if record_name == "pytorch":
 
record["depends"].append("cpuonly")
 
return index
return index
Loading