-
Notifications
You must be signed in to change notification settings - Fork 2
Home
Welcome to the Simple-StableLM-Chat wiki!
Open PowerShell
nvidia-smi
take note CUDA Version: 11.4
Invoke-WebRequest -Uri https://repo.anaconda.com/archive/Anaconda3-2021.05-Windows-x86_64.exe -OutFile Anaconda3-2021.05-Windows-x86_64.exe
.\Anaconda3-2021.05-Windows-x86_64.exe /S
wait a long time... anaconda is stupid to install
close PowerShell
launch Anaconda Prompt powershell
git clone https://github.com/astrobleem/Simple-StableLM-Chat.git
cd Simple-StableLM-Chat
conda create --name stablelm
conda activate stablelm
conda config --add channels pytorch
conda config --add channels conda-forge
#change cudatoolkit to match your installed version
conda install pytorch torchvision torchaudio cudatoolkit=11.4 -c pytorch -c nvidia
conda install transformers
pip install --upgrade urllib3
conda install --file requirements.txt
python -c "import torch; print(torch.cuda.is_available())"
python -c "import torch; print(torch.version)"
python .\chat.py