pytorch 安装

pytorch 安装

Step1: 在这个网站上找到自己电脑环境pytorch的安装方法 https://pytorch.org/get-started/locally/

我的cuda是11.0版本，采用pycharm自带的pip安装（有些博主是用的Conda安装的，并且配置的编译器是Conda）,我采用的编译器默认即可。

Step2: 在终端terminal输入（采用清华镜像）

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

注意在第2步下载过程中，下载进程可能会中途打断（停一下），到了最后几兆电脑会卡住（鼠标也没有反应），不要管它，等他自己变好（我就是看到电脑没有反应，按下了万能的电源键）。在第3步下载其他库的时候，也会暂停一下，不要管它，自己会慢慢变好。

Step3:检查pytorch是否安装成功

import torch
print(torch.cuda.is_available())
print(torch.__version__)

True
1.7.1+cu110