@@ -4,7 +4,8 @@ version = "0.1.0"
|
||||
edition = "2024"
|
||||
|
||||
[dependencies]
|
||||
burn = { version = "0.20.1", features = ["wgpu", "std", "fusion", "ndarray"] }
|
||||
burn = { version = "0.20.1", features = ["wgpu", "std", "fusion", "ndarray", "cuda"] }
|
||||
burn-autodiff = "0.20.1"
|
||||
|
||||
[dev-dependencies]
|
||||
rstest = "0.26.1"
|
||||
|
||||
Reference in New Issue
Block a user