Skip to content

Commit

Permalink
[doc] Add Mac M1 info in docs (#2040)
Browse files Browse the repository at this point in the history
  • Loading branch information
@xyang16
xyang16 authored Sep 27, 2022
1 parent cefbba7 commit 8fce4eb
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 4 deletions.
11 changes: 7 additions & 4 deletions docs/development/dependency_management.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ This document outlines how DJL manages its dependencies and how developers can f

We try our best to minimize the external dependencies. The core DJL library only has 4 external dependencies:

- [sl4fj-api](https://mvnrepository.com/artifact/ai.djl/api)
- [sl4fj-api](https://mvnrepository.com/artifact/org.slf4j/slf4j-api)
- [Gson](https://mvnrepository.com/artifact/com.google.code.gson/gson)
- [JNA](https://mvnrepository.com/artifact/net.java.dev.jna/jna)
- [apache-commons-compress](https://mvnrepository.com/artifact/org.apache.commons/commons-compress)
Expand Down Expand Up @@ -58,17 +58,20 @@ See [How to use DJL's BOM](../../bom/README.md#how-to-use-djls-bom) for detail.
| [ai.djl.pytorch](https://search.maven.org/search?q=g:ai.djl.pytorch) | [pytorch-engine](../../engines/pytorch/pytorch-engine/README.md#installation) | PyTorch engine adapter |
| | [pytorch-model-zoo](../../engines/pytorch/pytorch-model-zoo/README.md#installation) | Contains state of the art PyTorch torch script models |
| | [pytorch-native-cpu(osx-x86_64)](../../engines/pytorch/pytorch-engine/README.md#macos) | Contains PyTorch native library for macOS |
| | [pytorch-native-cpu(osx-aarch64)](../../engines/pytorch/pytorch-engine/README.md#macos-m1) | Contains PyTorch native library for macOS M1 |
| | [pytorch-native-cpu(win-x86_64)](../../engines/pytorch/pytorch-engine/README.md#windows-cpu) | Contains PyTorch native library for Windows |
| | [pytorch-native-cpu(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#linux-cpu) | Contains PyTorch native library for Linux |
| | [pytorch-native-cu113(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#linux-gpu) | Contains PyTorch native library for Linux with CUDA 11.3 |
| | [pytorch-native-cu113(win-x86_64)](../../engines/pytorch/pytorch-engine/README.md#windows-gpu) | Contains PyTorch native library for Windows with CUDA 11.3 |
| | [pytorch-native-cu116(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#linux-gpu) | Contains PyTorch native library for Linux with CUDA 11.6 |
| | [pytorch-native-cu116(win-x86_64)](../../engines/pytorch/pytorch-engine/README.md#windows-gpu) | Contains PyTorch native library for Windows with CUDA 11.6 |
| | [pytorch-native-cu102(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#linux-gpu) | Contains PyTorch native library for Linux with CUDA 10.2 |
| | [pytorch-native-cu102(win-x86_64)](../../engines/pytorch/pytorch-engine/README.md#windows-gpu) | Contains PyTorch native library for Windows with CUDA 10.2 |
| | [pytorch-native-cu113-precxx11(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#for-pre-cxx11-build) | Contains PyTorch native library for Linux with CUDA 11.3 |
| | [pytorch-native-cu116-precxx11(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#for-pre-cxx11-build) | Contains PyTorch native library for Linux with CUDA 11.6 |
| | [pytorch-native-cpu-precxx11(linux-x86_64)](../../engines/pytorch/pytorch-engine/README.md#for-pre-cxx11-build) | Contains PyTorch native library for centOS 7 and Ubuntu 14.04 |
| | [pytorch-native-cpu-precxx11(linux-aarch64)](../../engines/pytorch/pytorch-engine/README.md#for-aarch64-build) | Contains PyTorch native library for Linux ARM |
| | [pytorch-jni](../../engines/pytorch/pytorch-engine/README.md) | Contains PyTorch JNI native library |
| | pytorch-native-auto (deprecated) | No longer needed since DJL 0.15.0 |
| | pytorch-native-cu113(linux-x86_64) (deprecated) | Contains PyTorch native library for Linux with CUDA 11.3 |
| | pytorch-native-cu113(win-x86_64) (deprecated) | Contains PyTorch native library for Windows with CUDA 11.3 |
| | pytorch-native-cu111(linux-x86_64) (deprecated) | Contains PyTorch native library for Linux with CUDA 11.1 |
| | pytorch-native-cu111(win-x86_64) (deprecated) | Contains PyTorch native library for Windows with CUDA 11.1 |
| | pytorch-native-cu101(linux-x86_64) (deprecated) | Contains PyTorch native library <= 1.7.1 for Linux with CUDA 10.1 |
Expand Down
22 changes: 22 additions & 0 deletions engines/pytorch/pytorch-engine/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,28 @@ For macOS, you can use the following library:
</dependency>
```

### macOS M1
For macOS M1, you can use the following library:

- ai.djl.pytorch:pytorch-jni:1.12.1-0.19.0
- ai.djl.pytorch:pytorch-native-cpu:1.12.1:osx-aarch64

```xml
<dependency>
<groupId>ai.djl.pytorch</groupId>
<artifactId>pytorch-native-cpu</artifactId>
<classifier>osx-aarch64</classifier>
<version>1.12.1</version>
<scope>runtime</scope>
</dependency>
<dependency>
<groupId>ai.djl.pytorch</groupId>
<artifactId>pytorch-jni</artifactId>
<version>1.12.1-0.19.0</version>
<scope>runtime</scope>
</dependency>
```

### Linux
For the Linux platform, you can choose between CPU, GPU. If you have NVIDIA [CUDA](https://en.wikipedia.org/wiki/CUDA)
installed on your GPU machine, you can use one of the following library:
Expand Down

0 comments on commit 8fce4eb

Please sign in to comment.