From 21f6a73f66082f9ea68936a1ca20d99b73d8f2a0 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sat, 9 May 2020 20:09:18 +0300 Subject: [PATCH] Include CUDA version in the pytorch package fail error --- trains_agent/helper/package/pytorch.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/trains_agent/helper/package/pytorch.py b/trains_agent/helper/package/pytorch.py index addc606..7ed30b5 100644 --- a/trains_agent/helper/package/pytorch.py +++ b/trains_agent/helper/package/pytorch.py @@ -413,10 +413,10 @@ class PytorchRequirement(SimpleSubstitution): # else: # self.log.debug('Replacing requirement "%s" with %r', req, result) # return result + # self.log.debug( + # "Could not find Pytorch wheel in table, trying manually constructing URL" + # ) - self.log.debug( - "Could not find Pytorch wheel in table, trying manually constructing URL" - ) result = ok = None # try: # result, ok = self.get_url_for_platform(req) @@ -427,7 +427,7 @@ class PytorchRequirement(SimpleSubstitution): if result: self.log.debug("URL not found: {}".format(result)) exc = PytorchResolutionError( - "Could not find pytorch wheel URL for: {}".format(req) + "Could not find pytorch wheel URL for: {} with cuda {} support".format(req, self.cuda_version) ) # cancel exception chaining six.raise_from(exc, None)