build: prevent CUDA_HOME KeyError (#316)

address: #316 (comment)
dmlc · Nov 13, 2017 · 09b9718 · 09b9718
1 parent e083e5f
commit 09b9718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deps/build.jl b/deps/build.jl
@@ -63,7 +63,7 @@ if HAS_CUDA
   if HAS_CUDNN
     info("Found a CuDNN installation.")
   end
-  info("CUDA_HOME -> $(ENV["CUDA_HOME"])")
+  info("CUDA_HOME -> $(get(ENV, "CUDA_HOME", nothing))")
 else
   info("Did not find a CUDA installation, using CPU-only version of MXNet.")
 end