https://github.com/ssoudan/tch-m1

Example of tch-rs on M1
https://github.com/ssoudan/tch-m1

Last synced: 6 days ago
JSON representation

Example of tch-rs on M1

• Host: GitHub
• URL: https://github.com/ssoudan/tch-m1
• Owner: ssoudan
• Created: 2022-05-24T19:47:59.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-03-19T01:05:47.000Z (10 months ago)
• Last Synced: 2024-12-09T23:02:38.404Z (29 days ago)
• Language: Rust
• Size: 18.6 KB
• Stars: 51
• Watchers: 2
• Forks: 6
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Demo of tch-rs on M1

This repo demonstrate how to use [LaurentMazare/tch-rs](https://github.com/LaurentMazare/tch-rs) on M1.

## Steps to reproduce

- install micromamba with homebrew -- `brew install micromamba` - 

- create a new conda environment: `micromamba env create -f environment.yml`

- activate the new environment: `micromamba activate tch-rs-demo`

- create a symlink in this repo: `ln -sf ~/micromamba/envs/tch-rs-demo/lib/python3.10/site-packages/torch/ torch`

- run: `cargo run`

## More

- .cargo/config.toml is used to set envs for torch-sys to find the library and headers.