PyTorch Windows 安装教程:两行代码搞定 PyTorch 安装

打开 pytorch.org 官网,可以选择安装器,Python版本,CUDA 版本。

例如 Windows Pip Python3.6 CUDA 9.0 的安装命令为:

PyTorch Windows 安装教程:两行代码搞定 PyTorch 安装

pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision

原创文章,作者:pytorch,如若转载,请注明出处:http://pytorchchina.com/2018/12/11/pytorch-windows-install-1/

QR code