Windows AMD 安装 PyTorch

目录

Q &  A

前提

安装

 验证

---

Q &  A

Q：笔记本不是英伟达（NVIDIA）显卡，能安装PyTorch吗？

A： 可以！选择CPU为运算平台的PyTorch安装版本。（Compute Platform：CPU）

Q：选择以CPU为运算平台的PyTorch版本与GPU版本的在程序运行方面差别大吗？

A： 影响并不大！
从定位上看，pytorch和tensorflow不仅仅用于动辄几十层的vgg、res、dense各种net。还可以用于各种工业常用的机器学习算法。

尤其是tensorflow中包含的函数可以很方便的实现各种聚类、分类、回归等算法。比如，k均值、svm、贝叶斯网络等等。

更重要的是，很多研究人员根本没有足够的gpu，或者干脆不愿意去花时间配置cuda。复杂的配置必然对框架的普遍应用造成负面影响，参考caffee。

尤其是对于学生党，只有笔记本的情况下cpu版本是绝佳选择(笔记本跑gpu会热死)。

所以，cpu在很多情况下就够用，而且便于普及。
作者：机器学习入坑者
链接：https://www.zhihu.com/question/361894250/answer/1029343204
来源：知乎

前提

已经安装python和pip。

pip是Python官方建议的包安装工具。

win10 安装 Python 和 pip 详解_Passion 的博客-CSDN博客https://passion.blog.csdn.net/article/details/50470970

安装

1. 登录下方PyTorch官方网站，按照如图所示选择版本，并复制指令。

Start Locally | PyTorchhttps://pytorch.org/get-started/locally/

2. 使用键盘，WIN+R 打开运行窗口，键入cmd，打开命令行窗口。

3. Ctrl+V（或者点击鼠标右键）粘贴命令，点击回车键运行命令。出现类似于

Successfully installed torch-1.9.1 torchaudio-0.9.1 torchvision-0.10.1 typing-extensions-3.10.0.2

的文字，表示Pytorch库安装成功！

 验证

创建文件PyTorch_Test.py

import torch
print(torch.__version__)

运行该python程序文件，可查看pytorch版本。