- 博客(2)
- 资源 (1)
- 收藏
- 关注
原创 PyTorch学习
一、PyTorch简介 PyTorch是一个能在CPU和GPU上运行并解决各类深度学习问题的深度学习框架。可以将其看做是支持GPU计算和自动微分计算的Numpy库。 1.1 特点 易于使用的API Python的支持 动态计算图 部署简单 支持分布式训练 支持移动端 强大的生态系统 内置开放的神经网络交换协议(ONNX) 1.2 GPU使用测试 import torch # 以下代码只有在PyTorch GPU版本上才会执行 import time print(torch.__v.
2022-05-22 21:24:58
572
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人