Migrated install_requirements.bat to conda-forge

No more manually forcing CUDA to be 11, instead we use conda-forge.
This will pull down a lot more recent versions of pretty much everything, and fixes errors in the GPT-J models that cropped up on the older versions of the dependencies.
This commit is contained in:
henk717 2021-06-18 01:33:56 +02:00 committed by GitHub
parent 09f407c62e
commit d38384628c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -23,7 +23,7 @@ IF ERRORLEVEL 1 (
miniconda3.exe /S /InstallationType=JustMe /RegisterPython=0 /AddTopath=0 /NoScripts=1 /NoRegistry=1 /D=%~dp0\miniconda3 miniconda3.exe /S /InstallationType=JustMe /RegisterPython=0 /AddTopath=0 /NoScripts=1 /NoRegistry=1 /D=%~dp0\miniconda3
del miniconda3.exe del miniconda3.exe
call miniconda3\condabin\activate call miniconda3\condabin\activate
call conda install --all --no-shortcuts -y git pytorch tensorflow-gpu colorama Flask-SocketIO cudatoolkit=11 -c pytorch call conda install --all --no-shortcuts -y git pytorch tensorflow-gpu colorama Flask-SocketIO -c pytorch -c conda-forge
IF %M%==1 pip install git+https://github.com/finetuneanon/transformers@gpt-neo-localattention3 IF %M%==1 pip install git+https://github.com/finetuneanon/transformers@gpt-neo-localattention3
IF %M%==2 call conda install --no-shortcuts -y transformers -c huggingface IF %M%==2 call conda install --no-shortcuts -y transformers -c huggingface
call conda clean -a -y call conda clean -a -y