error[E0308]: mismatched types
  --> examples/tmp.rs:6:43
   |
6  |     let m: Model = Model::build_on_device(&dev);
   |                    ---------------------- ^^^^ expected struct `dfdx::tensor::Cpu`, found struct `dfdx::tensor::Cuda`
   |                    |
   |                    arguments to this function are incorrect
   |
   = note: expected reference `&dfdx::tensor::Cpu`
              found reference `&dfdx::tensor::Cuda`
note: associated function defined here
  --> /home/ubuntu/dfdx/src/nn/module.rs:49:8
   |
49 |     fn build_on_device(device: &D) -> Self::Output {
   |        ^^^^^^^^^^^^^^^

error[E0277]: the trait bound `dfdx::nn::Linear<5, 2>: dfdx::nn::BuildModule<dfdx::tensor::Cuda, f32>` is not satisfied
 --> examples/tmp.rs:5:39
  |
5 |     let m: Model = BuildModule::build(&dev);
  |                    ------------------ ^^^^ the trait `dfdx::nn::BuildModule<dfdx::tensor::Cuda, f32>` is not implemented for `dfdx::nn::Linear<5, 2>`
  |                    |
  |                    required by a bound introduced by this call
  |
  = help: the trait `dfdx::nn::BuildModule<D, f32>` is implemented for `dfdx::nn::Linear<I, O, D>`

For more information about this error, try `rustc --explain E0277`.