compile with cudart (pytorch#737)

kashif · soumith · commit 8d90ab2d9b73 · 2017-02-14T06:40:35.000+05:30
diff --git a/setup.py b/setup.py
@@ -275,6 +275,7 @@ def run(self):
     extra_link_args.append('-Wl,-rpath,' + cuda_lib_path)
     extra_compile_args += ['-DWITH_CUDA']
     extra_compile_args += ['-DCUDA_LIB_PATH=' + cuda_lib_path]
+    main_libraries += ['cudart']
     main_link_args += [THC_LIB, THCS_LIB, THCUNN_LIB]
     main_sources += [
         "torch/csrc/cuda/Module.cpp",
