getnode -p gpu # this also loads gcc
module add soft/nvidia/cuda/12.0
/usr/bin/python3.9 -m venv .env
source .env/bin/activate
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
cd lltm-extension
python setup.py install
python benchmark.py-
Notifications
You must be signed in to change notification settings - Fork 0
MartinBubel/torch-extension-playground
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Experimenting with torch cuda cpp tutorials
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published