diff --git a/README.md b/README.md index e285b792e..5797d1af7 100644 --- a/README.md +++ b/README.md @@ -15,10 +15,10 @@ Just run it in the terminal. Check latest cudnn and cudatoolkit version for your GPU device. ```bash - conda create -n deepfacelab -c main python=3.7 cudnn=7.6.5 cudatoolkit=10.1.243 - conda activate deepfacelab git clone --depth 1 https://github.com/nagadit/DeepFaceLab_Linux.git cd DeepFaceLab_Linux + conda env create -f environment.yml + conda activate deepfacelab git clone --depth 1 https://github.com/iperov/DeepFaceLab.git python -m pip install -r ./DeepFaceLab/requirements-cuda.txt ``` diff --git a/environment.yml b/environment.yml new file mode 100644 index 000000000..f3c8e60e5 --- /dev/null +++ b/environment.yml @@ -0,0 +1,10 @@ +name: deepfacelab +dependencies: + - python=3.7 + - cudnn=7.6.5 + - cudatoolkit=10.1.243 +variables: + DFL_PYTHON: python3.7 + DFL_WORKSPACE: "workspace/" + DFL_ROOT: "./" + DFL_SRC: "./DeepFaceLab" \ No newline at end of file diff --git a/scripts/env.sh b/scripts/env.sh index 152f583c2..648bda383 100644 --- a/scripts/env.sh +++ b/scripts/env.sh @@ -1,8 +1,5 @@ #!/usr/bin/env bash conda activate deepfacelab -cd .. -export DFL_PYTHON="python3.7" -export DFL_WORKSPACE="workspace/" if [ ! -d "$DFL_WORKSPACE" ]; then mkdir "$DFL_WORKSPACE" @@ -13,7 +10,4 @@ if [ ! -d "$DFL_WORKSPACE" ]; then mkdir "$DFL_WORKSPACE/data_dst/aligned" mkdir "$DFL_WORKSPACE/data_dst/aligned_debug" mkdir "$DFL_WORKSPACE/model" -fi - -export DFL_ROOT="./" -export DFL_SRC="./DeepFaceLab" \ No newline at end of file +fi \ No newline at end of file