We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 07c357e commit b2341afCopy full SHA for b2341af
ddddorc/README.MD
@@ -1,6 +1,16 @@
1
-# MacOS with MPS
2
-```
+# ddddorc trainer in MacOS with GPU(MPS)
3
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
13
14
15
16
+```
0 commit comments