-
Notifications
You must be signed in to change notification settings - Fork 2
Home
Welcome to the Simple-StableLM-Chat wiki!
Open PowerShell PS D:\downloads> 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