diff --git a/src/model.rs b/src/model.rs
index f1a0b7be188cca7d66ef0741aa5d615337f83977..424a984cc62b4d7dd9a7ce3ee06b3faf9d13240b 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.