Unverified Commit 76699b4a authored by Tim Dettmers's avatar Tim Dettmers Committed by GitHub
Browse files

Merge pull request #37 from tomaarsen/hotfix/colab_just_cpu

Perform check using implicit list length
parents 7740c6e9 58aa7c53
......@@ -103,7 +103,7 @@ def get_compute_capability(cuda):
None.
"""
ccs = get_compute_capabilities(cuda)
if ccs is not None:
if ccs:
# TODO: handle different compute capabilities; for now, take the max
return ccs[-1]
return None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment