windows下安装pytorch

提示：可以直接用第3步，输入命令，一步到位！

1.装好python和pip

2.进pytorch官网 https://pytorch.org/   

  3.命令行输入

pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html  -i https://pypi.tuna.tsinghua.edu.cn/simple/ 

注意：-i : 换清华源，速度快

看网上教程很多，复杂又难用，各种bug，忍不了了，自己发个博客。至于CUDA版本选10.2，这个和nvidia显卡驱动版本有关，百度一下吧，也可以换成9.2试试。

 

补充：tensorflow安装命令，python用3.5版本

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow