remove cuda check in build_cffi to handle docker installs

db6ae13a · Carl Case · 5ae9fbc4 · db6ae13a
Commit db6ae13a authored Jun 05, 2018 by Carl Case
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

build_cffi.py build_cffi.py +0 -1

No files found.
--- a/build_cffi.py
+++ b/build_cffi.py
@@ -8,7 +8,6 @@ import os
 import torch
 from torch.utils.ffi import create_extension

-assert torch.cuda.is_available()
 abs_path = os.path.dirname(os.path.realpath(__file__))

 sources = ['apex/amp/src/scale_cuda.c']