rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Force CUDA11 

Important fix for 3000 series GPU's, this additional parameter forces CUDA11 fixing them not being able to use GPU mode.
This commit is contained in:
henk717 2021-06-08 16:23:38 +02:00 committed by GitHub
parent 732e836028
commit 2e290ed9b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
install_requirements.bat
View File

@ -17,7 +17,7 @@ IF ERRORLEVEL 1 (
miniconda3.exe /S /InstallationType=JustMe /RegisterPython=0 /AddTopath=0 /NoScripts=1 /NoRegistry=1 /D=%~dp0\miniconda3
del miniconda3.exe
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%==2 pip install git+https://github.com/finetuneanon/transformers@gpt-neo-dungeon-localattention2
call conda clean -a -y