[init]

ssoudan · ssoudan · commit 626993bf9bac · 2022-05-24T12:48:13.000-07:00
diff --git a/.cargo/config.toml b/.cargo/config.toml
@@ -0,0 +1,3 @@
+[env]
+LIBTORCH = { value = "torch" , relative = true }
+DYLD_LIBRARY_PATH = { value = "torch/lib" , relative = true }
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,3 @@
+/target
+/torch
+/.idea/
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -0,0 +1,9 @@
+[package]
+name = "tch-m1"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+tch = {  version = "0.7.2", default-features = false }
diff --git a/README.md b/README.md
@@ -0,0 +1,16 @@
+# Demo of tch-rs on M1
+
+## Steps
+
+- install miniforge with homebrew -- See  https://naolin.medium.com/conda-on-m1-mac-with-miniforge-bbc4e3924f2b
+- create a new conda environment: `conda create -n tch-rs-demo python=3.8`
+- activate the new environment: `conda activate tch-rs-demo`
+- install pytorch: `conda install pytorch`
+- create a symlink in this repo: `ln -s /opt/homebrew/Caskroom/miniforge/base/envs/tch-rs-demo/lib/python3.8/site-packages/torch/ torch`
+- run: `cargo run`
+
+## Details
+
+- .cargo/config.toml is used to set envs for torch-sys to find the library and headers.
+- Tensor::print() segfaults.
+
diff --git a/src/main.rs b/src/main.rs

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+[env]`
	`2`	`+LIBTORCH = { value = "torch" , relative = true }`
	`3`	`+DYLD_LIBRARY_PATH = { value = "torch/lib" , relative = true }`