如何安装pytorch?Facebook开发的深度学习框架
发布于 作者:苏南大叔 来源:程序如此灵动~
既然TensorFlow没有想象中的那么厚重,那么pytorch也必然是可以运行起来的。本文描述在windows环境下安装pytorch的经验文字。整篇文章非常简单,就是简要的描述了一个命令,并做了个简单的验证。

苏南大叔的“程序如此灵动”博客,记录苏南大叔的代码编程经验总结。测试环境:win10,python@3.12.9,pytorch@2.6.0。
pytorch
PyTorch是一个开源的深度学习框架,由Facebook的人工智能研究团队开发。其官方网站:

pytorch是经常和tensorflow并列的深度学习框架。参考文章:
安装pytorch
pytorch是基于python的torch实现。虽然包名是pytorch,但是安装命令是:
pip install torch
又是一个包名和安装名不一样的例子...
可以使用下面的命令,验证是否安装完全。
pip show torch
基础代码
下面的代码就是个pytorch的helloworld代码,简单的很。
import torch
# 输出 PyTorch 版本号
print("PyTorch 版本号:", torch.__version__)
# 创建一个张量
hello_tensor = torch.tensor([1, 2, 3, 4, 5])
print("Hello, PyTorch!")
print(hello_tensor)
结语
匆匆的就引入了pytorch,是因为苏南大叔打算以后的机器学习领域的代码,都使用sklearn、pytorch、tensorflow三个代码来实现一遍。这样也有个对比性。侧面证明三者确实是有相同覆盖的领域。