5,323
edits
PyTorch: Difference between revisions
no edit summary
(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== | ||