PyTorch: Difference between revisions
No edit summary |
|||
| Line 6: | Line 6: | ||
# If using conda, python 3.5+, and CUDA 10.0 (+ compatible cudnn) | # If using conda, python 3.5+, and CUDA 10.0 (+ compatible cudnn) | ||
conda install pytorch torchvision cudatoolkit=10. | conda install pytorch torchvision cudatoolkit=10.0 -c pytorch | ||
</syntaxhighlight> | </syntaxhighlight> | ||
==Usage== | ==Usage== | ||
Revision as of 20:12, 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.0 -c pytorch