Skip to content

Commit 6e65588

Browse files
committed
Updating the ROCm PIP case to install torchvision
- Adding ninja to speed up the build
1 parent b30d8bd commit 6e65588

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

assets/quick-start-module.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ function commandMessage(key) {
231231
"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",
232232

233233
"stable,pip,linux,rocm4.0,python":
234-
"pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html",
234+
"pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html<br />pip install ninja && pip install 'git+https://github.com/pytorch/vision.git@v0.9.0'",
235235

236236
"stable,pip,windows,accnone,python":
237237
"pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html",

0 commit comments

Comments
 (0)