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