浏览代码

Update README.MD

MoeClub 4 月之前
父节点
当前提交
f368f92307
共有 1 个文件被更改,包括 1 次插入1 次删除
  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