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