Merge pull request #40 from henk717/patch-3
Bugfix: Force CUDA11 in new installer for 3000 series GPU's
This commit is contained in:
commit
b4fef1de1b
|
@ -17,7 +17,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 -c pytorch
|
call conda install --all --no-shortcuts -y git pytorch tensorflow-gpu colorama Flask-SocketIO cudatoolkit=11 -c pytorch
|
||||||
IF %M%==1 call conda install --no-shortcuts -y transformers -c huggingface
|
IF %M%==1 call conda install --no-shortcuts -y transformers -c huggingface
|
||||||
IF %M%==2 pip install git+https://github.com/finetuneanon/transformers@gpt-neo-dungeon-localattention2
|
IF %M%==2 pip install git+https://github.com/finetuneanon/transformers@gpt-neo-dungeon-localattention2
|
||||||
call conda clean -a -y
|
call conda clean -a -y
|
||||||
|
|
Loading…
Reference in New Issue