Skip to content

Commit 8c49b51

Browse files
soumithsoumith
committed
Add CUDA 10.2 and make it the default
1 parent 211c383 commit 8c49b51

File tree

2 files changed

+62
-5
lines changed

2 files changed

+62
-5
lines changed

‎_includes/quick_start_local.html

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,13 +85,16 @@
8585
<div class="col-md-12 title-block mobile-heading">
8686
<div class="option-text">CUDA</div>
8787
</div>
88-
<div class="col-md-4 option block version" id="cuda9.2">
88+
<div class="col-md-3 option block version" id="cuda9.2">
8989
<div class="option-text">9.2</div>
9090
</div>
91-
<div class="col-md-4 option block version selected" id="cuda10.1">
91+
<div class="col-md-3 option block version" id="cuda10.1">
9292
<div class="option-text">10.1</div>
9393
</div>
94-
<div class="col-md-4 option block version" id="cudanone">
94+
<div class="col-md-3 option block version selected" id="cuda10.2">
95+
<div class="option-text">10.2</div>
96+
</div>
97+
<div class="col-md-3 option block version" id="cudanone">
9598
<div class="option-text">None</div>
9699
</div>
97100
</div>

‎assets/quick-start-module.js

Lines changed: 56 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ var supportedOperatingSystems = new Map([
66
]);
77

88
var opts = {
9-
cuda: 'cuda10.1',
9+
cuda: 'cuda10.2',
1010
os: getAnchorSelectedOS() || getDefaultSelectedOS(),
1111
pm: 'conda',
1212
language: 'python',
@@ -181,6 +181,9 @@ function commandMessage(key) {
181181
"stable,conda,linux,cuda10.1,python":
182182
"conda install pytorch torchvision cudatoolkit=10.1 -c pytorch",
183183

184+
"stable,conda,linux,cuda10.2,python":
185+
"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch",
186+
184187
"stable,conda,linux,cudanone,python":
185188
"conda install pytorch torchvision cpuonly -c pytorch",
186189

@@ -190,6 +193,9 @@ function commandMessage(key) {
190193
"stable,conda,macos,cuda10.1,python":
191194
"conda install pytorch torchvision -c pytorch<br /># MacOS Binaries dont support CUDA, install from source if CUDA is needed",
192195

196+
"stable,conda,macos,cuda10.2,python":
197+
"conda install pytorch torchvision -c pytorch<br /># MacOS Binaries dont support CUDA, install from source if CUDA is needed",
198+
193199
"stable,conda,macos,cudanone,python":
194200
"conda install pytorch torchvision -c pytorch",
195201

@@ -199,6 +205,9 @@ function commandMessage(key) {
199205
"stable,conda,windows,cuda10.1,python":
200206
"conda install pytorch torchvision cudatoolkit=10.1 -c pytorch",
201207

208+
"stable,conda,windows,cuda10.2,python":
209+
"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch",
210+
202211
"stable,conda,windows,cudanone,python":
203212
"conda install pytorch torchvision cpuonly -c pytorch",
204213

@@ -208,6 +217,9 @@ function commandMessage(key) {
208217
"stable,pip,macos,cuda10.1,python":
209218
"pip install torch torchvision<br /># MacOS Binaries dont support CUDA, install from source if CUDA is needed",
210219

220+
"stable,pip,macos,cuda10.2,python":
221+
"pip install torch torchvision<br /># MacOS Binaries dont support CUDA, install from source if CUDA is needed",
222+
211223
"stable,pip,macos,cudanone,python": "pip install torch torchvision",
212224

213225
"stable,pip,linux,cudanone,python":
@@ -217,6 +229,9 @@ function commandMessage(key) {
217229
"pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html",
218230

219231
"stable,pip,linux,cuda10.1,python":
232+
"pip install torch==1.4.0+cu101 torchvision==0.5.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html",
233+
234+
"stable,pip,linux,cuda10.2,python":
220235
"pip install torch torchvision",
221236

222237
"stable,pip,windows,cudanone,python":
@@ -226,6 +241,9 @@ function commandMessage(key) {
226241
"pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html",
227242

228243
"stable,pip,windows,cuda10.1,python":
244+
"pip install torch==1.4.0+cu101 torchvision==0.5.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html",
245+
246+
"stable,pip,windows,cuda10.2,python":
229247
"pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html",
230248

231249
"stable,libtorch,linux,cudanone,cplusplus":
@@ -235,7 +253,10 @@ function commandMessage(key) {
235253
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu92/libtorch-shared-with-deps-1.4.0%2Bcu92.zip'>https://download.pytorch.org/libtorch/cu92/libtorch-shared-with-deps-1.4.0%2Bcu92.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu92/libtorch-cxx11-abi-shared-with-deps-1.4.0%2Bcu92.zip'>https://download.pytorch.org/libtorch/cu92/libtorch-cxx11-abi-shared-with-deps-1.4.0%2Bcu92.zip</a>",
236254

237255
"stable,libtorch,linux,cuda10.1,cplusplus":
238-
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.4.0.zip'>https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.4.0.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.4.0.zip'>https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.4.0.zip</a>",
256+
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.4.0%2Bcu101.zip'>https://download.pytorch.org/libtorch/cu101/libtorch-shared-with-deps-1.4.0%2Bcu101.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.4.0%2Bcu101.zip'>https://download.pytorch.org/libtorch/cu101/libtorch-cxx11-abi-shared-with-deps-1.4.0%2Bcu101.zip</a>",
257+
258+
"stable,libtorch,linux,cuda10.2,cplusplus":
259+
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.4.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.4.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.4.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-1.4.0.zip</a>",
239260

240261
"stable,libtorch,macos,cudanone,cplusplus":
241262
"MacOS binaries do not support Java. Support is only available for Linux. Download here for C++: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip </a>",
@@ -246,6 +267,9 @@ function commandMessage(key) {
246267
"stable,libtorch,macos,cuda10.1,cplusplus":
247268
"MacOS binaries do not support Java. Support is only available for Linux. MacOS binaries do not support CUDA. Download CPU libtorch here for C++: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip </a>",
248269

270+
"stable,libtorch,macos,cuda10.2,cplusplus":
271+
"MacOS binaries do not support Java. Support is only available for Linux. MacOS binaries do not support CUDA. Download CPU libtorch here for C++: <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip'> https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.4.0.zip </a>",
272+
249273
"stable,libtorch,windows,cudanone,cplusplus":
250274
"Windows binaries do not support Java. Support is only available for Linux. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.4.0.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.4.0.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.4.0.zip'>https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-1.4.0.zip</a>",
251275

@@ -255,12 +279,18 @@ function commandMessage(key) {
255279
"stable,libtorch,windows,cuda10.1,cplusplus":
256280
"Windows binaries do not support Java. Support is only available for Linux. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-1.4.0.zip'>https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-1.4.0.zip</a><br/><br> Download here for C++ (Debug version): <br/><a href='https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-debug-1.4.0.zip'>https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-debug-1.4.0.zip</a>",
257281

282+
"stable,libtorch,windows,cuda10.2,cplusplus":
283+
"Windows binaries do not support Java. Support is only available for Linux. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.4.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.4.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.4.0.zip'>https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-debug-1.4.0.zip</a>",
284+
258285
"preview,conda,linux,cuda9.2,python":
259286
"conda install pytorch torchvision cudatoolkit=9.2 -c pytorch-nightly",
260287

261288
"preview,conda,linux,cuda10.1,python":
262289
"conda install pytorch torchvision cudatoolkit=10.1 -c pytorch-nightly",
263290

291+
"preview,conda,linux,cuda10.2,python":
292+
"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly",
293+
264294
"preview,conda,linux,cudanone,python":
265295
"conda install pytorch torchvision cpuonly -c pytorch-nightly",
266296

@@ -270,6 +300,9 @@ function commandMessage(key) {
270300
"preview,conda,macos,cuda10.1,python":
271301
"conda install pytorch torchvision -c pytorch-nightly",
272302

303+
"preview,conda,macos,cuda10.2,python":
304+
"conda install pytorch torchvision -c pytorch-nightly",
305+
273306
"preview,conda,macos,cudanone,python":
274307
"conda install pytorch torchvision -c pytorch-nightly",
275308

@@ -279,6 +312,9 @@ function commandMessage(key) {
279312
"preview,conda,windows,cuda10.1,python":
280313
"conda install pytorch torchvision cudatoolkit=10.1 -c pytorch-nightly -c defaults -c conda-forge",
281314

315+
"preview,conda,windows,cuda10.2,python":
316+
"conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly -c defaults -c conda-forge",
317+
282318
"preview,conda,windows,cudanone,python":
283319
"conda install pytorch torchvision cpuonly -c pytorch-nightly -c defaults -c conda-forge",
284320

@@ -288,6 +324,9 @@ function commandMessage(key) {
288324
"preview,pip,macos,cuda10.1,python":
289325
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html<br /># On MacOS, we provide CPU-only packages, CUDA functionality is not provided",
290326

327+
"preview,pip,macos,cuda10.2,python":
328+
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html<br /># On MacOS, we provide CPU-only packages, CUDA functionality is not provided",
329+
291330
"preview,pip,macos,cudanone,python":
292331
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html",
293332

@@ -300,6 +339,9 @@ function commandMessage(key) {
300339
"preview,pip,linux,cuda10.1,python":
301340
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html",
302341

342+
"preview,pip,linux,cuda10.2,python":
343+
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html",
344+
303345
"preview,pip,windows,cudanone,python":
304346
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html",
305347

@@ -309,6 +351,9 @@ function commandMessage(key) {
309351
"preview,pip,windows,cuda10.1,python":
310352
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html",
311353

354+
"preview,pip,windows,cuda10.2,python":
355+
"pip install numpy<br />pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html",
356+
312357
"preview,libtorch,linux,cudanone,cplusplus":
313358
"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>",
314359

@@ -318,6 +363,9 @@ function commandMessage(key) {
318363
"preview,libtorch,linux,cuda10.1,cplusplus":
319364
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu101/libtorch-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu101/libtorch-shared-with-deps-latest.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu101/libtorch-cxx11-abi-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu101/libtorch-cxx11-abi-shared-with-deps-latest.zip</a>",
320365

366+
"preview,libtorch,linux,cuda10.2,cplusplus":
367+
"Download here (Pre-cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu102/libtorch-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu102/libtorch-shared-with-deps-latest.zip</a><br/><br> Download here (cxx11 ABI): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu102/libtorch-cxx11-abi-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu102/libtorch-cxx11-abi-shared-with-deps-latest.zip</a>",
368+
321369
"preview,libtorch,macos,cudanone,cplusplus":
322370
"MacOS binaries do not support Java. Support is only available for Linux. Download here for C++: <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>",
323371

@@ -327,6 +375,9 @@ function commandMessage(key) {
327375
"preview,libtorch,macos,cuda10.1,cplusplus":
328376
"MacOS binaries do not support Java. Support is only available for Linux. MacOS binaries do not support CUDA. Download CPU libtorch here for C++: <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>",
329377

378+
"preview,libtorch,macos,cuda10.2,cplusplus":
379+
"MacOS binaries do not support Java. Support is only available for Linux. MacOS binaries do not support CUDA. Download CPU libtorch here for C++: <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>",
380+
330381
"preview,libtorch,windows,cudanone,cplusplus":
331382
"Windows binaries do not support Java. Support is only available for Linux. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/nightly/cpu/libtorch-win-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cpu/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/cpu/libtorch-win-shared-with-deps-debug-latest.zip'>https://download.pytorch.org/libtorch/nightly/cpu/libtorch-win-shared-with-deps-debug-latest.zip</a>",
332383

@@ -335,6 +386,9 @@ function commandMessage(key) {
335386

336387
"preview,libtorch,windows,cuda10.1,cplusplus":
337388
"Windows binaries do not support Java. Support is only available for Linux. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu101/libtorch-win-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu101/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/cu101/libtorch-win-shared-with-deps-debug-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu101/libtorch-win-shared-with-deps-debug-latest.zip</a>",
389+
390+
"preview,libtorch,windows,cuda10.2,cplusplus":
391+
"Windows binaries do not support Java. Support is only available for Linux. Download here for C++ (Release version): <br/><a href='https://download.pytorch.org/libtorch/nightly/cu102/libtorch-win-shared-with-deps-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu102/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/cu102/libtorch-win-shared-with-deps-debug-latest.zip'>https://download.pytorch.org/libtorch/nightly/cu102/libtorch-win-shared-with-deps-debug-latest.zip</a>",
338392
};
339393

340394
if (!object.hasOwnProperty(key)) {

0 commit comments

Comments
 (0)