pytorch搭建卷积神经网络【第一课

文章目录

• 前言
• 导入torch
• 查看torch版本
• 创建tensor
• tensor运算
• tensor转移至GPU 需要配置cudnn
• 总结

---

前言

例如：随着人工智能的不断发展，机器学习这门技术也越来越重要，很多人都开启了学习机器学习，本文就介绍了机器学习的基础内容。

源码路径：
github：https://github.com/2012Netsky/pytorch_cnn/blob/main/1_making_sure_things_work.py

---

导入torch

#!/usr/bin/env python
# coding: utf-8

# 导入torch
import torch

查看torch版本

# 查看torch版本
torch.version.__version__

创建tensor

# 创建tensor
a = torch.ones(3,3)

# 
b = torch.ones(3,3)

tensor运算

# tensor运算
a + b

tensor转移至GPU 需要配置cudnn

# tensor转移至GPU 需要配置cudnn
a = a.to('cuda')
b = b.to('cuda')

---

总结

提示：这里对文章进行总结：
例如：以上就是今天要讲的内容，本文仅仅简单介绍了pandas的使用，而pandas提供了大量能使我们快速便捷地处理数据的函数和方法。