Upload sdwui-start-util.dev.ipynb

sdwui-start-util.dev.ipynb

@@ -1419,8 +1419,8 @@
     "        mkdirs(f'{_install_path}/{_ui_dir_name}/venv')\n",
     "#         run('python3.10 -m venv venv',cwd=f'{_install_path}/{_ui_dir_name}')\n",
     "        run(f'tar -xf {venvPath} -C ./venv',cwd=f'{_install_path}/{_ui_dir_name}')\n",
-    "    run(f'rm -f {_install_path}/{_ui_dir_name}/venv/bin/pip*')\n",
-    "    run(f'rm -f {_install_path}/{_ui_dir_name}/venv/bin/python*')\n",
+    "        run(f'rm -f {_install_path}/{_ui_dir_name}/venv/bin/pip*')\n",
+    "        run(f'rm -f {_install_path}/{_ui_dir_name}/venv/bin/python*')\n",
     "    venv.create(f'{_install_path}/{_ui_dir_name}/venv')\n",
     "    if not Path(f'{_install_path}/{_ui_dir_name}/venv/bin/pip').exists():\n",
     "        run('curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py')\n",
@@ -1448,7 +1448,7 @@
     "    #     run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install Pillow==9.5.0''')\n",
     "    #     run(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install basicsr''')\n",
     "        \n",
-    "    get_ipython().system(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install basicsr''')\n",
+    "    get_ipython().system(f'''{_install_path}/{_ui_dir_name}/venv/bin/python3 -m pip install basicsr wrapt''')\n",
     "    \n",
     "    envInstalled = True\n",
     "    print('安装需要的python环境 完成')\n",
@@ -1504,8 +1504,7 @@
     "        os.chdir(f'{_install_path}/{_ui_dir_name}')\n",
     "        root_path = _sub_path[i]\n",
     "        if root_path.endswith('/'): root_path = root_path[:-1]\n",
-    "        if torch.cuda.device_count() == 2 and not _multi_case:\n",
-    "            os.environ['CUDA_VISIBLE_DEVICES']='0,1'\n",
+    "        if torch.cuda.device_count() == 1 or not _multi_case:\n",
     "            get_ipython().system(f'''venv/bin/python3 launch.py --port {str(_server_port+1+i)} --subpath={_sub_path[i]}''')\n",
     "        else:    \n",
     "            get_ipython().system(f'''venv/bin/python3 launch.py --device-id={i} --port {str(_server_port+1+i)} --subpath={_sub_path[i]}''')\n",