Change to uninstall torch related packages when prepare_relase

synthesis/prepare_enunu_release.py

@@ -12,6 +12,7 @@
 from typing import List
 
 KEEP_LATEST_PACKAGES = ['pip', 'setuptools', 'wheel', 'utaupy']
+TORCH_PACKAGES= ['torch', 'torchaudio', 'torchvision']
 REMOVE_LIST = ['__pycache__', '.mypy']
 PYTHON_DIR = 'python-3.8.10-embed-amd64'
 
@@ -24,6 +25,13 @@ def pip_install_upgrade(python_exe: str, packages: List[str]):
             '--no-warn-script-location'] + packages
     subprocess.run(args, check=True)
 
+def pip_uninstall_torch(python_exe: str, packages: List[str]):
+    """
+    pythonのパッケージを更新する
+    """
+    args = [python_exe, '-m', 'pip', 'uninstall'] + packages
+    subprocess.run(args, check=True)
+
 
 def remove_cache_files(path_dir, remove_list):
     """
@@ -127,6 +135,7 @@ def main():
     python_exe = join(python_dir, 'python.exe')
     print(f'Upgrading packages of {python_dir} (this may take some minutes)')
     pip_install_upgrade(python_exe, KEEP_LATEST_PACKAGES)
+    pip_uninstall_torch(python_exe, TORCH_PACKAGES)    
     print()
 
     # Pythonの実行ファイルをコピーする