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.1 -c pytorch
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
</syntaxhighlight>
</syntaxhighlight>


==Usage==
==Usage==
Retrieved from "https://wiki.davidl.me/view/PyTorch"