![PyTorch深度学习与企业级项目实战](https://wfqqreader-1252317822.image.myqcloud.com/cover/140/52842140/b_52842140.jpg)
上QQ阅读APP看书,第一时间看更新
2.2 在Windows环境下安装CPU版的PyTorch
PytTorch是基于Python开发的,首先需要安装Python,Python的安装很简单,这里不再赘述。而Windows用户能直接通过conda、pip和源码编译三种方式来安装PyTorch。
打开PyTorch官网(https://pytorch.org/),在主页中根据自己的计算机选择Linux、Mac或Windows系统,如图2-8所示,系统将给出对应的安装命令语句,比如这里为pip3 install torch torchvision torchaudio。
![](https://epubservercos.yuewen.com/E2A8D2/31397752303496206/epubprivate/OEBPS/Images/Figure-P28_681.jpg?sign=1739151444-J2JB76F2V1S6yrOgTh6KwVXh1ZkIyfdu-0-cc7452bb5e7b01867161c66317edc19d)
图2-8
通过安装命令安装PyTorch 2.1.0,结果如下:
![](https://epubservercos.yuewen.com/E2A8D2/31397752303496206/epubprivate/OEBPS/Images/Figure-P28_684.jpg?sign=1739151444-dJ9XK5UmCTNfm6V1adgIVKNywiXzantX-0-6a64cb779f0421ed7c0124990ee32e40)
![](https://epubservercos.yuewen.com/E2A8D2/31397752303496206/epubprivate/OEBPS/Images/Figure-P29_988.jpg?sign=1739151444-LH1mipWfW94e2MqYTfgv1UlRJ5zNH7IC-0-0b2c8e8a129ec51e5479f16d43f5f5f6)
验证PyTorch是否安装成功,执行如下命令,注意命令中的双下画线:
print(torch.__version__) print(torch.version.cuda) print(torch.cuda.is_available())
命令执行结果如下:
![](https://epubservercos.yuewen.com/E2A8D2/31397752303496206/epubprivate/OEBPS/Images/Figure-P30_993.jpg?sign=1739151444-HePw7znKCaLkSKmsIxSJZDtc5yBR3WWd-0-c1d61a62cabeb6df0c81814a8ac173b6)
如果没有报错,则说明PyTorch安装成功。