From ec3f374d74738978c11dfdf417673d51dd9216bb Mon Sep 17 00:00:00 2001 From: Yukio Siraichi Date: Thu, 14 Dec 2023 20:42:18 -0300 Subject: [PATCH] Temporarily compile PyTorch with CUDA support. --- .circleci/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.circleci/build.sh b/.circleci/build.sh index 68e81a5436c..626bb226cbe 100755 --- a/.circleci/build.sh +++ b/.circleci/build.sh @@ -42,7 +42,7 @@ apply_patches python -c "import fcntl; fcntl.fcntl(1, fcntl.F_SETFL, 0)" # We always build PyTorch without CUDA support. -export USE_CUDA=0 +export USE_CUDA=1 python setup.py install sccache --show-stats