5,323
edits
PyTorch: Difference between revisions
→Installation
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== | ||