ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-03-03 10:42:05 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Include CUDA version in the pytorch package fail error

Browse Source
This commit is contained in:
allegroai 2020-05-09 20:09:18 +03:00
parent 77c4c79a2f
commit 21f6a73f66
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
trains_agent/helper/package/pytorch.py
View File

@ -413,10 +413,10 @@ class PytorchRequirement(SimpleSubstitution):
# else: # else:
# self.log.debug('Replacing requirement "%s" with %r', req, result) # self.log.debug('Replacing requirement "%s" with %r', req, result)
# return 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 result = ok = None
# try: # try:
# result, ok = self.get_url_for_platform(req) # result, ok = self.get_url_for_platform(req)
@ -427,7 +427,7 @@ class PytorchRequirement(SimpleSubstitution):
if result: if result:
self.log.debug("URL not found: {}".format(result)) self.log.debug("URL not found: {}".format(result))
exc = PytorchResolutionError( 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 # cancel exception chaining
six.raise_from(exc, None) six.raise_from(exc, None)