From 4e414312c8936c9a6e17e8017a684a9e8b1fac90 Mon Sep 17 00:00:00 2001
From: Robin Lobel <divide@divideconcept.net>
Date: Wed, 2 Mar 2022 18:54:28 +0100
Subject: [PATCH] allow PyTorch 1.9 as minimum requirement, fix PyTorch install
 on mac

---
 torchstudio/pythoncheck.py   | 2 +-
 torchstudio/pythoninstall.py | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/torchstudio/pythoncheck.py b/torchstudio/pythoncheck.py
index 898efee..fc8d741 100644
--- a/torchstudio/pythoncheck.py
+++ b/torchstudio/pythoncheck.py
@@ -36,7 +36,7 @@
         else:
             from importlib.metadata import version
         pytorch_version=tuple(int(i) for i in version('torch').split('.')[:2])
-        min_pytorch_version=(1,9) if args.remote else (1,10) #1.9 required for torch.package support, 1.10 required for stable torch.fx and profile-directed typing in torchscript
+        min_pytorch_version=(1,9) #1.9 required for torch.package support, 1.10 preferred for stable torch.fx and profile-directed typing in torchscript
         if pytorch_version<min_pytorch_version:
             print("Error: PyTorch "+str(min_pytorch_version[0])+"."+str(min_pytorch_version[1])+" minimum is required.", file=sys.stderr)
             print("This environment has PyTorch "+str(pytorch_version[0])+"."+str(pytorch_version[1])+".", file=sys.stderr)
diff --git a/torchstudio/pythoninstall.py b/torchstudio/pythoninstall.py
index 5e14875..910edcb 100644
--- a/torchstudio/pythoninstall.py
+++ b/torchstudio/pythoninstall.py
@@ -25,7 +25,8 @@
     else:
         conda_install="pytorch torchvision torchaudio cpuonly datasets scipy pandas matplotlib-base python-graphviz paramiko pysoundfile"
 elif sys.platform.startswith('darwin'):
-    conda_install="nomkl pytorch torchvision torchaudio datasets scipy pandas matplotlib-base python-graphviz paramiko"
+    # force a pytorch/mkl version, because pytorch 1.10.2+ depends on mkl 2022 which is incompatible with Rosetta 2 in M1 macs
+    conda_install="pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 mkl==2021.4.0 datasets scipy pandas matplotlib-base python-graphviz paramiko"
 elif sys.platform.startswith('linux'):
     if args.gpu:
         conda_install="pytorch torchvision torchaudio cudatoolkit=11.3 datasets scipy pandas matplotlib-base python-graphviz paramiko"
@@ -40,7 +41,7 @@
 print(conda_install)
 print("")
 
-# channels: pytorch for pytorch torchvision torchaudio, nvidia for cudatoolkit=11.1 on Linux, huggingface for datasets(+huggingface_hub), conda-forge for everything else except anaconda for python-graphviz
+# channels: pytorch for pytorch torchvision torchaudio, nvidia for cudatoolkit=11.1 on Linux, huggingface for datasets(+huggingface_hub), conda-forge for everything else except anaconda for python-graphviz and ngam for osx-arm64 torchaudio
 conda_install+=" -c pytorch -c nvidia -c huggingface -c conda-forge -c anaconda"
 
 # https://stackoverflow.com/questions/41767340/using-conda-install-within-a-python-script