PyTorch: Difference between revisions

PyTorch (view source)

7 bytes added , 23 April

5,323

edits

@@ Line 2: / Line 2: @@
 ==Installation==
-See [https://pytorch.org/get-started/locally/ PyTorch Getting Started]
+See [https://pytorch.org/get-started/locally/ PyTorch Getting Started] and [https://pytorch.org/get-started/previous-versions/ PyTorch Previous Versions]
-<syntaxhighlight lang="bash">
-# If using conda, python 3.5+, and CUDA 10.0 (+ compatible cudnn)
+I recommend using the conda installation method since it is paired with the correct version of cuda.
-conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
-</syntaxhighlight>
 ==Getting Started==