Skip to content

Commit

Permalink
Add CUDA to package version during setup
Browse files Browse the repository at this point in the history
  • Loading branch information
casper-hansen committed Oct 29, 2023
1 parent 1c6e7ae commit 96e4e25
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,13 @@
os.environ["CC"] = "g++"
os.environ["CXX"] = "g++"

try:
CUDA_VERSION = "".join(os.environ.get("CUDA_VERSION", torch.version.cuda).split("."))
except Exception as ex:
raise RuntimeError("Your system must have an Nvidia GPU for installing AutoAWQ")

common_setup_kwargs = {
"version": "0.1.5",
"version": f"0.1.5+cu{CUDA_VERSION}",
"name": "autoawq",
"author": "Casper Hansen",
"license": "MIT",
Expand Down

0 comments on commit 96e4e25

Please sign in to comment.