Skip to content

Commit 256aa18

Browse files
committed
assets: Add CUDA 11 sections to quick start
Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
1 parent 230b71e commit 256aa18

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

assets/quick-start-module.js

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,9 @@ function commandMessage(key) {
184184
"stable,conda,linux,cuda10.2,python":
185185
"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch",
186186

187+
"stable,conda,linux,cuda11.0,python":
188+
"conda install pytorch torchvision cudatoolkit=11.0 -c pytorch",
189+
187190
"stable,conda,linux,cudanone,python":
188191
"conda install pytorch torchvision cpuonly -c pytorch",
189192

@@ -208,6 +211,9 @@ function commandMessage(key) {
208211
"stable,conda,windows,cuda10.2,python":
209212
"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch",
210213

214+
"stable,conda,windows,cuda11.0,python":
215+
"conda install pytorch torchvision cudatoolkit=11.0 -c pytorch",
216+
211217
"stable,conda,windows,cudanone,python":
212218
"conda install pytorch torchvision cpuonly -c pytorch",
213219

@@ -237,6 +243,9 @@ function commandMessage(key) {
237243
"stable,pip,linux,cuda10.2,python":
238244
"pip install torch torchvision",
239245

246+
"stable,pip,linux,cuda11.0,python":
247+
"pip install torch==1.7.0+cu110 torchvision==0.8.0+cu110 -f https://download.pytorch.org/whl/torch_stable.html",
248+
240249
"stable,pip,windows,cudanone,python":
241250
"pip install torch==1.7.0+cpu torchvision==0.8.0+cpu -f https://download.pytorch.org/whl/torch_stable.html",
242251

@@ -246,6 +255,9 @@ function commandMessage(key) {
246255
"stable,pip,windows,cuda10.2,python":
247256
"pip install torch===1.7.0 torchvision===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html",
248257

258+
"stable,pip,windows,cuda11.0,python":
259+
"pip install torch===1.7.0+cu110 torchvision===0.8.0+cu110 -f https://download.pytorch.org/whl/torch_stable.html",
260+
249261
"stable,libtorch,linux,cudanone,cplusplus":
250262
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.7.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-1.7.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.7.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.7.0%2Bcpu.zip</a>",
251263

@@ -258,6 +270,9 @@ function commandMessage(key) {
258270
"stable,libtorch,linux,cuda10.2,cplusplus":
259271
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.7.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.7.0.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.7.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.7.0.zip</a>",
260272

273+
"stable,libtorch,linux,cuda11.0,cplusplus":
274+
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0.zip'>https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu110/libtorch-cxx11-abi-shared-with-deps-1.7.0.zip'>https://download.pytorch.org/libtorch/cu110/libtorch-cxx11-abi-shared-with-deps-1.7.0.zip</a>",
275+
261276
"stable,libtorch,macos,cudanone,cplusplus":
262277
"Download here: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.7.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.7.0.zip </a>",
263278

@@ -271,7 +286,7 @@ function commandMessage(key) {
271286
"MacOS binaries do not support CUDA. Download CPU libtorch here: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.7.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.7.0.zip </a>",
272287

273288
"stable,libtorch,macos,cuda11.0,cplusplus":
274-
"MacOS binaries do not support CUDA. Download CPU libtorch here: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.6.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.6.0.zip </a>",
289+
"MacOS binaries do not support CUDA. Download CPU libtorch here: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.7.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.7.0.zip </a>",
275290

276291
"stable,libtorch,windows,cudanone,cplusplus":
277292
"Windows binaries do not support Java. 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.7.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.7.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.7.0%2Bcpu.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-1.7.0%2Bcpu.zip</a>",
@@ -282,6 +297,9 @@ function commandMessage(key) {
282297
"stable,libtorch,windows,cuda10.2,cplusplus":
283298
"Windows binaries do not support Java. Support is only available for Linux and MacOS. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.7.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.7.0.zip</a><br/><br> Download here for C++ (Debug version): <br/><a href='https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-debug-1.7.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-debug-1.7.0.zip</a>",
284299

300+
"stable,libtorch,windows,cuda11.0,cplusplus":
301+
"Windows binaries do not support Java. Support is only available for Linux and MacOS. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cu110/libtorch-win-shared-with-deps-1.7.0.zip'>https://download.pytorch.org/libtorch/cu110/libtorch-win-shared-with-deps-1.7.0.zip</a><br/><br> Download here for C++ (Debug version): <br/><a href='https://download.pytorch.org/libtorch/cu110/libtorch-win-shared-with-deps-debug-1.7.0.zip'>https://download.pytorch.org/libtorch/cu110/libtorch-win-shared-with-deps-debug-1.7.0.zip</a>",
302+
285303
"preview,conda,linux,cuda9.2,python":
286304
"conda install pytorch torchvision cudatoolkit=9.2 -c pytorch-nightly",
287305

0 commit comments

Comments
 (0)