Upgrade to PyTorch 2.3.0

a45acf9f · Woosuk Kwon · 733117ea · a45acf9f · a45acf9f
Commit a45acf9f authored May 06, 2024 by Woosuk Kwon
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

build.sh build.sh +1 -1

setup.py setup.py +1 -1

No files found.
--- a/build.sh
+++ b/build.sh
@@ -2,7 +2,7 @@

 eval "$(conda shell.bash hook)"

-PYTORCH_VERSION="2.2.1"
+PYTORCH_VERSION="2.3.0"

 for PYTHON_VERSION in 38 39 310 311; do
    source ~/.bashrc;

--- a/setup.py
+++ b/setup.py
@@ -263,7 +263,7 @@ class NinjaBuildExtension(BuildExtension):
        super().__init__(*args, **kwargs)


-PYTORCH_VERSION = "2.2.1"
+PYTORCH_VERSION = "2.3.0"
 CUDA_VERSION = "12.1"

 setup(