Skip to content

Commit b2341af

Browse files
authored
Update README.MD
1 parent 07c357e commit b2341af

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

ddddorc/README.MD

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,16 @@
1-
# MacOS with MPS
2-
```
1+
# ddddorc trainer in MacOS with GPU(MPS)
32

43
```
4+
git clone https://github.com/sml2h3/dddd_trainer.git
5+
cd dddd_trainer
6+
7+
pip3 install -r requirements.txt
8+
sed -i '' "s/[[:space:]]*device = torch\.device('cuda:{}'/ if hasattr(torch.backends, "mps") and torch.backends.mps.is_available() and torch.backends.mps.is_built():\n device = torch.device('mps:{}'.format(str(gpu_id)))\n else:\n device = torch.device('cuda:{}'/" nets/__init__.py
9+
10+
pip3 uninstall onnxruntime
11+
pip3 install onnxruntime-silicon
12+
pip3 install torch torchvision torchaudio
513
614
15+
16+
```

0 commit comments

Comments
 (0)