Skip to content

Commit ac6fe08

Browse files
atalmanatalman
authored
Enable cuda 12.4 build for release - Update matrix for release 2.4 (#1683)
* Enable cuda 12.4 build for release * update
1 parent 2561b6c commit ac6fe08

File tree

3 files changed

+260
-6
lines changed

3 files changed

+260
-6
lines changed

‎assets/quick-start-module.js

+3-3
Large diffs are not rendered by default.

‎published_versions.json

+256-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
{
2-
"latest_stable": "2.3.1",
2+
"latest_stable": "2.4.0",
33
"latest_lts": "lts-1.8.2",
44
"versions": {
55
"preview": {
@@ -4588,6 +4588,261 @@
45884588
}
45894589
}
45904590
}
4591+
},
4592+
"2.4.0": {
4593+
"linux": {
4594+
"pip": {
4595+
"accnone": {
4596+
"note": null,
4597+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu"
4598+
},
4599+
"cuda.x": {
4600+
"note": null,
4601+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118"
4602+
},
4603+
"cuda.y": {
4604+
"note": null,
4605+
"command": "pip3 install torch torchvision torchaudio"
4606+
},
4607+
"cuda.z": {
4608+
"note": null,
4609+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124"
4610+
},
4611+
"rocm5.x": {
4612+
"note": null,
4613+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1"
4614+
}
4615+
},
4616+
"conda": {
4617+
"cuda.x": {
4618+
"note": null,
4619+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia"
4620+
},
4621+
"cuda.y": {
4622+
"note": null,
4623+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia"
4624+
},
4625+
"cuda.z": {
4626+
"note": null,
4627+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia"
4628+
},
4629+
"rocm5.x": {
4630+
"note": "<b>NOTE:</b> Conda packages are not currently available for ROCm, please use pip instead<br />",
4631+
"command": null
4632+
},
4633+
"accnone": {
4634+
"note": null,
4635+
"command": "conda install pytorch torchvision torchaudio cpuonly -c pytorch"
4636+
}
4637+
},
4638+
"libtorch": {
4639+
"accnone": {
4640+
"note": null,
4641+
"versions": {
4642+
"Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.4.0%2Bcpu.zip",
4643+
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.4.0%2Bcpu.zip"
4644+
}
4645+
},
4646+
"cuda.x": {
4647+
"note": null,
4648+
"versions": {
4649+
"Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cu118/libtorch-shared-with-deps-2.4.0%2Bcu118.zip",
4650+
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.4.0%2Bcu118.zip"
4651+
}
4652+
},
4653+
"cuda.y": {
4654+
"note": null,
4655+
"versions": {
4656+
"Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.4.0%2Bcu121.zip",
4657+
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.4.0%2Bcu121.zip"
4658+
}
4659+
},
4660+
"cuda.z": {
4661+
"note": null,
4662+
"versions": {
4663+
"Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/cu124/libtorch-shared-with-deps-2.4.0%2Bcu124.zip",
4664+
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/cu124/libtorch-cxx11-abi-shared-with-deps-2.4.0%2Bcu124.zip"
4665+
}
4666+
},
4667+
"rocm5.x": {
4668+
"note": null,
4669+
"versions": {
4670+
"Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/rocm6.1/libtorch-shared-with-deps-2.4.0%2Brocm6.1.zip",
4671+
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/rocm6.1/libtorch-cxx11-abi-shared-with-deps-2.4.0%2Brocm6.1.zip"
4672+
}
4673+
}
4674+
}
4675+
},
4676+
"macos": {
4677+
"pip": {
4678+
"cuda.x": {
4679+
"note": "# CUDA is not available on MacOS, please use default package",
4680+
"command": "pip3 install torch torchvision torchaudio",
4681+
"default": true
4682+
},
4683+
"cuda.y": {
4684+
"note": "# CUDA is not available on MacOS, please use default package",
4685+
"command": "pip3 install torch torchvision torchaudio",
4686+
"default": true
4687+
},
4688+
"cuda.z": {
4689+
"note": "# CUDA is not available on MacOS, please use default package",
4690+
"command": "pip3 install torch torchvision torchaudio",
4691+
"default": true
4692+
},
4693+
"rocm5.x": {
4694+
"note": "# ROCm is not available on MacOS, please use default package",
4695+
"command": "pip3 install torch torchvision torchaudio",
4696+
"default": true
4697+
},
4698+
"accnone": {
4699+
"note": null,
4700+
"command": "pip3 install torch torchvision torchaudio"
4701+
}
4702+
},
4703+
"conda": {
4704+
"cuda.x": {
4705+
"note": "# CUDA is not available on MacOS, please use default package",
4706+
"command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch",
4707+
"default": true
4708+
},
4709+
"cuda.y": {
4710+
"note": "# CUDA is not available on MacOS, please use default package",
4711+
"command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch",
4712+
"default": true
4713+
},
4714+
"cuda.z": {
4715+
"note": "# CUDA is not available on MacOS, please use default package",
4716+
"command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch",
4717+
"default": true
4718+
},
4719+
"rocm5.x": {
4720+
"note": "# ROCm is not available on MacOS, please use default package",
4721+
"command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch",
4722+
"default": true
4723+
},
4724+
"accnone": {
4725+
"note": null,
4726+
"command": "conda install pytorch::pytorch torchvision torchaudio -c pytorch"
4727+
}
4728+
},
4729+
"libtorch": {
4730+
"accnone": {
4731+
"note": null,
4732+
"versions": {
4733+
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.0.zip"
4734+
}
4735+
},
4736+
"cuda.x": {
4737+
"note": null,
4738+
"default": true,
4739+
"versions": {
4740+
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.0.zip"
4741+
}
4742+
},
4743+
"cuda.y": {
4744+
"note": null,
4745+
"default": true,
4746+
"versions": {
4747+
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.0.zip"
4748+
}
4749+
},
4750+
"cuda.z": {
4751+
"note": null,
4752+
"default": true,
4753+
"versions": {
4754+
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.0.zip"
4755+
}
4756+
},
4757+
"rocm5.x": {
4758+
"note": null,
4759+
"default": true,
4760+
"versions": {
4761+
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.4.0.zip"
4762+
}
4763+
}
4764+
}
4765+
},
4766+
"windows": {
4767+
"pip": {
4768+
"accnone": {
4769+
"note": null,
4770+
"command": "pip3 install torch torchvision torchaudio"
4771+
},
4772+
"cuda.x": {
4773+
"note": null,
4774+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118"
4775+
},
4776+
"cuda.y": {
4777+
"note": null,
4778+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121"
4779+
},
4780+
"cuda.z": {
4781+
"note": null,
4782+
"command": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124"
4783+
},
4784+
"rocm5.x": {
4785+
"note": "<b>NOTE:</b> ROCm is not available on Windows",
4786+
"command": null
4787+
}
4788+
},
4789+
"conda": {
4790+
"cuda.x": {
4791+
"note": null,
4792+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia"
4793+
},
4794+
"cuda.y": {
4795+
"note": null,
4796+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia"
4797+
},
4798+
"cuda.z": {
4799+
"note": null,
4800+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia"
4801+
},
4802+
"rocm5.x": {
4803+
"note": "<b>NOTE:</b> ROCm is not available on Windows",
4804+
"command": null
4805+
},
4806+
"accnone": {
4807+
"note": null,
4808+
"command": "conda install pytorch torchvision torchaudio cpuonly -c pytorch"
4809+
}
4810+
},
4811+
"libtorch": {
4812+
"accnone": {
4813+
"note": null,
4814+
"versions": {
4815+
"Download here (Release version):": "https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-2.4.0%2Bcpu.zip",
4816+
"Download here (Debug version):": "https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-2.4.0%2Bcpu.zip"
4817+
}
4818+
},
4819+
"cuda.x": {
4820+
"note": null,
4821+
"versions": {
4822+
"Download here (Release version):": "https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-2.4.0%2Bcu118.zip",
4823+
"Download here (Debug version):": "https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-debug-2.4.0%2Bcu118.zip"
4824+
}
4825+
},
4826+
"cuda.y": {
4827+
"note": null,
4828+
"versions": {
4829+
"Download here (Release version):": "https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-2.4.0%2Bcu121.zip",
4830+
"Download here (Debug version):": "https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-debug-2.4.0%2Bcu121.zip"
4831+
}
4832+
},
4833+
"cuda.z": {
4834+
"note": null,
4835+
"versions": {
4836+
"Download here (Release version):": "https://download.pytorch.org/libtorch/cu124/libtorch-win-shared-with-deps-2.4.0%2Bcu124.zip",
4837+
"Download here (Debug version):": "https://download.pytorch.org/libtorch/cu124/libtorch-win-shared-with-deps-debug-2.4.0%2Bcu124.zip"
4838+
}
4839+
},
4840+
"rocm5.x": {
4841+
"note": "<b>NOTE:</b> ROCm is not available on Windows",
4842+
"versions": null
4843+
}
4844+
}
4845+
}
45914846
}
45924847
}
45934848
}

‎scripts/gen_quick_start_module.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ class OperatingSystem(Enum):
4444
"accnone": ("cpu", ""),
4545
"cuda.x": ("cuda", "11.8"),
4646
"cuda.y": ("cuda", "12.1"),
47+
"cuda.z": ("cuda", "12.4"),
4748
"rocm5.x": ("rocm", "6.0")
4849
}
4950
}
@@ -107,8 +108,6 @@ def update_versions(versions, release_matrix, release_version):
107108

108109
if release_version != "nightly":
109110
version = release_matrix[OperatingSystem.LINUX.value][0]["stable_version"]
110-
# temporary change until release 2.4.0 - to generate cuda.x, cuda.y, cuda.z
111-
template = "2.3.0"
112111
if version not in versions["versions"]:
113112
versions["versions"][version] = copy.deepcopy(versions["versions"][template])
114113
versions["latest_stable"] = version

0 commit comments

Comments
 (0)