From 2e290ed9b95f019841f2f7ea1ee00ba6540292ef Mon Sep 17 00:00:00 2001
From: henk717 <henk717@msn.com>
Date: Tue, 8 Jun 2021 16:23:38 +0200
Subject: [PATCH] Force CUDA11

Important fix for 3000 series GPU's, this additional parameter forces CUDA11 fixing them not being able to use GPU mode.
---
 install_requirements.bat | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/install_requirements.bat b/install_requirements.bat
index 79736652..645283fb 100644
--- a/install_requirements.bat
+++ b/install_requirements.bat
@@ -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