From e83726bb596fb6e58a017f42686534639cc6dbc7 Mon Sep 17 00:00:00 2001 From: Iago Bonnici <iago.bonnici@umontpellier.fr> Date: Mon, 11 Mar 2024 16:47:06 +0100 Subject: [PATCH] Stop requesting cuda device. --- src/model.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model.rs b/src/model.rs index f1a0b7b..424a984 100644 --- a/src/model.rs +++ b/src/model.rs @@ -39,7 +39,7 @@ pub fn optimize_likelihood( start: &Parameters<f64>, ) -> (Parameters<f64>, f64) { // Choose in-memory data location. - let (kind, device) = (Kind::Double, Device::cuda_if_available()); + let (kind, device) = (Kind::Double, Device::Cpu); let vars = nn::VarStore::new(device); // Construct the likelihood function from fixed data and initial trainable parameters. -- GitLab