pytorch-gpu版安装教程

按照我的方法安装玩tensorflow-gpu后,接下来就是安装pytoch-gpu版了

打开https://pytorch.org/get-started/locally/

pytorch-gpu版安装教程_第1张图片

pip install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

pytorch-gpu版安装教程_第2张图片

pip install torchvision  -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

pytorch-gpu版安装教程_第3张图片

 

你可能感兴趣的:(系统安装环境搭建)