如何安装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
三个代码来实现一遍。这样也有个对比性。侧面证明三者确实是有相同覆盖的领域。


