Skip to content

Adding few updates for ROCm #613

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Mar 12, 2021
Merged

Adding few updates for ROCm #613

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2a68a46
Adding few updates for ROCm
pruthvistony Mar 8, 2021
39ba26d
Changes based on the review comments for ROCm case
pruthvistony Mar 9, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 11 additions & 11 deletions assets/quick-start-module.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,7 @@ function commandMessage(key) {
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio -c pytorch<br /># MacOS Binaries dont support CUDA, install from source if CUDA is needed",

"stable,conda,macos,rocm4.0,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio -c pytorch<br /># ROCm is not available on MacOS",
"<b>NOTE:</b> ROCm is not available on MacOS",

"stable,conda,macos,accnone,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio -c pytorch",
Expand All @@ -205,7 +205,7 @@ function commandMessage(key) {
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch",

"stable,conda,windows,rocm4.0,python":
"<b>NOTE:</b> ROCm is not available on Windows<br />",
"<b>NOTE:</b> ROCm is not available on Windows",

"stable,conda,windows,accnone,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio cpuonly -c pytorch",
Expand All @@ -217,7 +217,7 @@ function commandMessage(key) {
"pip install torch torchvision torchaudio<br /># MacOS Binaries dont support CUDA, install from source if CUDA is needed",

"stable,pip,macos,rocm4.0,python":
"<b>NOTE:</b> ROCm is not available on MacOS<br />",
"<b>NOTE:</b> ROCm is not available on MacOS",

"stable,pip,macos,accnone,python": "pip install torch torchvision torchaudio",

Expand All @@ -243,7 +243,7 @@ function commandMessage(key) {
"pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html",

"stable,pip,windows,rocm4.0,python":
"<b>NOTE:</b> ROCm is not available on Windows<br />",
"<b>NOTE:</b> ROCm is not available on Windows",

"stable,libtorch,linux,accnone,cplusplus":
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.8.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.8.0%2Bcpu.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.8.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.8.0%2Bcpu.zip</a>",
Expand All @@ -267,7 +267,7 @@ function commandMessage(key) {
"MacOS binaries do not support CUDA. Download CPU libtorch here: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.8.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.8.0.zip </a>",

"stable,libtorch,macos,rocm4.0,cplusplus":
"ROCm is not available on MacOS",
"<b>NOTE:</b> ROCm is not available on MacOS",

"stable,libtorch,windows,accnone,cplusplus":
"Windows stable binaries do not support Java, but nighty binaries do. Support is only available for Linux and MacOS. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.8.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.8.0%2Bcpu.zip</a><br/><br> Download here for C++ (Debug version): <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-1.8.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-1.8.0%2Bcpu.zip</a>",
Expand All @@ -279,7 +279,7 @@ function commandMessage(key) {
"Windows stable binaries do not support Java, but nighty binaries do. Support is only available for Linux and MacOS. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cu111/libtorch-win-shared-with-deps-1.8.0%2Bcu111.zip'>https://download.pytorch.org/libtorch/cu111/libtorch-win-shared-with-deps-1.8.0%2Bcu111.zip</a><br/><br> Download here for C++ (Debug version): <br/><a href='https://download.pytorch.org/libtorch/cu111/libtorch-win-shared-with-deps-debug-1.8.0%2Bcu111.zip'>https://download.pytorch.org/libtorch/cu111/libtorch-win-shared-with-deps-debug-1.8.0%2Bcu111.zip</a>",

"stable,libtorch,windows,rocm4.0,cplusplus":
"<b>NOTE:</b> ROCm is not available on Windows<br />",
"<b>NOTE:</b> ROCm is not available on Windows",

"preview,conda,linux,cuda10.2,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch-nightly",
Expand All @@ -300,7 +300,7 @@ function commandMessage(key) {
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio -c pytorch-nightly",

"preview,conda,macos,rocm4.0,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio -c pytorch-nightly",
"<b>NOTE:</b> ROCm is not available on MacOS",

"preview,conda,macos,accnone,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio -c pytorch-nightly",
Expand All @@ -312,7 +312,7 @@ function commandMessage(key) {
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-nightly",

"preview,conda,windows,rocm4.0,python":
"<b>NOTE:</b> ROCm is not available on Windows<br />",
"<b>NOTE:</b> ROCm is not available on Windows",

"preview,conda,windows,accnone,python":
"<b>NOTE:</b> Python 3.9 users will need to add '-c=conda-forge' for installation<br />conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly",
Expand Down Expand Up @@ -351,7 +351,7 @@ function commandMessage(key) {
"pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu111/torch_nightly.html",

"preview,pip,windows,rocm4.0,python":
"<b>NOTE:</b> ROCm is not available on Windows<br />",
"<b>NOTE:</b> ROCm is not available on Windows",

"preview,libtorch,linux,accnone,cplusplus":
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cpu/libtorch-cxx11-abi-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cpu/libtorch-cxx11-abi-shared-with-deps-latest.zip</a>",
Expand All @@ -363,7 +363,7 @@ function commandMessage(key) {
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu111/libtorch-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu111/libtorch-shared-with-deps-latest.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu111/libtorch-cxx11-abi-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu111/libtorch-cxx11-abi-shared-with-deps-latest.zip</a>",

"preview,libtorch,linux,rocm4.0,cplusplus":
"<b>NOTE:</b> Conda packages are not currently available for ROCm, please use pip instead<br />",
"LibTorch binaries are not available for ROCm, please build it from source",

"preview,libtorch,macos,accnone,cplusplus":
"Download here: <br/><a href='https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-latest.zip'> https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-latest.zip </a>",
Expand All @@ -387,7 +387,7 @@ function commandMessage(key) {
"Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu111/libtorch-win-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu111/libtorch-win-shared-with-deps-latest.zip</a><br/><br> Download here for C++ (Debug version): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu111/libtorch-win-shared-with-deps-debug-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu111/libtorch-win-shared-with-deps-debug-latest.zip</a>",

"preview,libtorch,windows,rocm4.0,cplusplus":
"<b>NOTE:</b> ROCm is not available on Windows<br />",
"<b>NOTE:</b> ROCm is not available on Windows",
};

if (!object.hasOwnProperty(key)) {
Expand Down