From e3026a53a3c71d433f1b2ed7af2272941fc834a2 Mon Sep 17 00:00:00 2001 From: Teddy Koker Date: Tue, 23 Mar 2021 17:41:58 -0400 Subject: [PATCH 1/2] docker fix --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index def77c7..776e819 100644 --- a/setup.py +++ b/setup.py @@ -5,6 +5,7 @@ from functools import lru_cache from subprocess import DEVNULL, call +import torch from setuptools import setup from torch.utils import cpp_extension @@ -14,7 +15,7 @@ def cuda_toolkit_available(): # https://github.com/idiap/fast-transformers/blob/master/setup.py try: call(["nvcc"], stdout=DEVNULL, stderr=DEVNULL) - return True + return True and torch.cuda.is_available() except FileNotFoundError: return False From df7f96617d6011bc6167b176e5eb4dc988c5cdf4 Mon Sep 17 00:00:00 2001 From: Teddy Koker Date: Tue, 23 Mar 2021 20:01:26 -0400 Subject: [PATCH 2/2] version 0.1.1 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 776e819..281ef78 100644 --- a/setup.py +++ b/setup.py @@ -51,7 +51,7 @@ def ext_modules(): setup( name="torchsort", - version="0.1.0", + version="0.1.1", description="Differentiable sorting and ranking in PyTorch", author="Teddy Koker", url="https://github.com/teddykoker/torchsort",