PyTorch: Difference between revisions
(Created page with "PyTorch is a popular machine learning library developed by Facebook ==Usage==") |
No edit summary |
||
| Line 1: | Line 1: | ||
PyTorch is a popular machine learning library developed by Facebook | PyTorch is a popular machine learning library developed by Facebook | ||
==Installation== | |||
See [https://pytorch.org/get-started/locally/ PyTorch Getting Started] | |||
<syntaxhighlight lang="bash"> | |||
# If using conda, python 3.5+, and CUDA 10.0 (+ compatible cudnn) | |||
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch | |||
</syntaxhighlight> | |||
==Usage== | ==Usage== | ||
Revision as of 20:11, 2 January 2020
PyTorch is a popular machine learning library developed by Facebook
Installation
# If using conda, python 3.5+, and CUDA 10.0 (+ compatible cudnn)
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch