- 从零开始大模型开发与微调:基于PyTorch与ChatGLM
- 王晓华
- 137字
- 2024-12-31 17:37:13
2.2.3 PyTorch 2.0小练习:Hello PyTorch
恭喜读者,到这里我们已经完成了PyTorch 2.0的安装。打开CMD窗口依次输入如下命令即可验证安装是否成功:
import torch result = torch.tensor(1) + torch.tensor(2.0) result
运行结果如图2-22所示。
图2-22 验证安装是否成功
或者打开前面安装的PyCharm IDE,新建一个项目,再新建一个hello_pytorch.py文件,输入如下代码:
import torch result = torch.tensor(1) + torch.tensor(2.0) print(result)
最终结果请读者自行验证。