From 0e3df1d810a9e8855608ac6aa4a406a88d614c4e Mon Sep 17 00:00:00 2001 From: Roman S Samarev Date: Sat, 29 Oct 2022 20:06:31 -0700 Subject: [PATCH] added check to prevent CUDA driver activation if CUDA is not functional --- src/ONNXRunTime.jl | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/ONNXRunTime.jl b/src/ONNXRunTime.jl index ec2451f..752c101 100644 --- a/src/ONNXRunTime.jl +++ b/src/ONNXRunTime.jl @@ -15,7 +15,9 @@ include("capi.jl") include("highlevel.jl") function __init__() - @require CUDA="052768ef-5323-5732-b1bb-66c8b64840ba" include("cuda.jl") + @require CUDA="052768ef-5323-5732-b1bb-66c8b64840ba" begin + CUDA.functional() && include("cuda.jl") + end end end #module