From 0b082e90a19244a4a917c4f60ebb63973053c41c Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Wed, 28 Oct 2020 19:03:31 +0100
Subject: [PATCH] Make pytorch packages to depend on the cpuonly package
---
deps/repodata-patches/gen_patch_json.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/deps/repodata-patches/gen_patch_json.py b/deps/repodata-patches/gen_patch_json.py
index 02ea91a8..edc95fcd 100644
--- a/deps/repodata-patches/gen_patch_json.py
+++ b/deps/repodata-patches/gen_patch_json.py
@@ -110,6 +110,11 @@ def _gen_new_index(repodata, packages_key):
if parse_version(record["version"]) <= parse_version("4.1.0"):
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
--
GitLab