Skip to content

GitLab

Commit a45acf9f authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Upgrade to PyTorch 2.3.0

parent 733117ea
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
build.sh
View file @ a45acf9f
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
eval "$(conda shell.bash hook)" eval "$(conda shell.bash hook)"
PYTORCH_VERSION="2.2.1" PYTORCH_VERSION="2.3.0"
for PYTHON_VERSION in 38 39 310 311; do for PYTHON_VERSION in 38 39 310 311; do
source ~/.bashrc; source ~/.bashrc;
......
setup.py
View file @ a45acf9f
...@@ -263,7 +263,7 @@ class NinjaBuildExtension(BuildExtension): ...@@ -263,7 +263,7 @@ class NinjaBuildExtension(BuildExtension):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
PYTORCH_VERSION = "2.2.1" PYTORCH_VERSION = "2.3.0"
CUDA_VERSION = "12.1" CUDA_VERSION = "12.1"
setup( setup(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment