Browse Source

Update README.MD

MoeClub 4 months ago
parent
commit
f368f92307
1 changed files with 1 additions and 1 deletions
  1. 1 1
      ddddorc/README.MD

+ 1 - 1
ddddorc/README.MD

@@ -5,7 +5,7 @@ git clone https://github.com/sml2h3/dddd_trainer.git
 cd dddd_trainer
 
 pip3 install -r requirements.txt
-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
+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
 
 pip3 uninstall onnxruntime
 pip3 install onnxruntime-silicon