Skip to content

Instantly share code, notes, and snippets.

@acidsound

acidsound/torch.bat

Created September 17, 2023 18:04
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save acidsound/dbba9f1528fbea5085a5b36fa4ea5c7c to your computer and use it in GitHub Desktop.
Save acidsound/dbba9f1528fbea5085a5b36fa4ea5c7c to your computer and use it in GitHub Desktop.
Download ZIP
windows pytorch 2.0.1 설정
Raw
torch.bat
pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
@acidsound
Copy link
Author

torchaudio 필요시

pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

@acidsound
Copy link
Author

2.1.0 version
pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

@acidsound
Copy link
Author

xformer 의존성 cu121

pip install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 --extra-index-url https://download.pytorch.org/whl/cu121

@acidsound
Copy link
Author

windows 에서 bitsandbytes 문제 해결

you should do this:

pip uninstall bitsandbytes
pip install https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.41.0-py3-none-win_amd64.whl
Why? Because you've downloaded a version of Bitsandbytes that isn't suitable for Windows, these tips are all for the Linux version to go wrong and will mislead you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment