diff --git a/Cargo.toml b/Cargo.toml
index 7c565f3..5e21079 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -15,7 +15,7 @@ default = ["tch"]
 download-libtorch = ["tch/download-libtorch"]
 
 [dependencies]
-tch = { version = "0.16", optional = true }
+tch = { version = "0.17", optional = true }
 einops-macros = { path = "einops-macros", version = "0.1.0-alpha.2" }
 
 [package.metadata.docs.rs]