From 9c9e9607eab78aa2310c2003ae375cad91c82bca Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 10 Jul 2022 17:23:23 +0200 Subject: [PATCH] Update m1_setup.sh --- m1_setup.sh | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/m1_setup.sh b/m1_setup.sh index df75ea2..20ed0d6 100644 --- a/m1_setup.sh +++ b/m1_setup.sh @@ -5,6 +5,9 @@ export HOMEBREW_CORE_GIT_REMOTE="https://github.com/Homebrew/homebrew-core" # p eval "$(/opt/homebrew/bin/brew shellenv)" brew update --force --quiet chmod -R go-w "$(brew --prefix)/share/zsh" +export OPENBLAS=$(/opt/homebrew/bin/brew --prefix openblas) +export CFLAGS="-falign-functions=8 ${CFLAGS}" + brew install wget brew install unzip @@ -14,6 +17,9 @@ conda activate torch-gpu conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 -c pytorch conda install -c conda-forge jupyter jupyterlab +python3 -m pip install --upgrade pip +python3 -m pip install insightface==0.2.1 onnx imageio scikit-learn scikit-image moviepy + #wget -P ./arcface_model https://github.com/neuralchen/SimSwap/releases/download/1.0/arcface_checkpoint.tar #wget https://github.com/neuralchen/SimSwap/releases/download/1.0/checkpoints.zip #unzip ./checkpoints.zip -d ./checkpoints @@ -21,21 +27,17 @@ conda install -c conda-forge jupyter jupyterlab #wget --no-check-certificate "https://sh23tw.dm.files.1drv.com/y4mmGiIkNVigkSwOKDcV3nwMJulRGhbtHdkheehR5TArc52UjudUYNXAEvKCii2O5LAmzGCGK6IfleocxuDeoKxDZkNzDRSt4ZUlEt8GlSOpCXAFEkBwaZimtWGDRbpIGpb_pz9Nq5jATBQpezBS6G_UtspWTkgrXHHxhviV2nWy8APPx134zOZrUIbkSF6xnsqzs3uZ_SEX_m9Rey0ykpx9w" -O antelope.zip #unzip ./antelope.zip -d ./insightface_func/models/ -python3 -m pip install insightface==0.2.1 onnxruntime moviepy -python3 -m pip install onnxruntime-gpu -python3 -m pip install insightface==0.2.1 onnxruntime moviepy python3 -m pip install googledrivedownloader python3 -m pip install imageio==2.4.1 -python3 -m pip install numpy==1.23.1 +python3 -m pip install numpy python3 -m pip install torch python3 -m pip install image python3 -m pip install timm python3 -m pip install PlL -python3 -m pip install h5py==3.7.0 +python3 -m pip install h5py -python3 m1_test.py for i in `seq 1 6`; do -python3 train.py --name simswap224_test --batchSize 8000000000000 --gpu_ids 0 --dataset crop_224/$i.jpg +python3 m1_test.py done conda deactivate