PyTorch 学习笔记
学习记录 PyTorch,记录开发学习过程
PyTorch 实操笔记
1. 简单介绍
- PyTorch 是一个深度学习框架,主要用于研究和生产环境
- 由Meta AI(Facebook)人工智能研究小组开发的一种基于Lua编写的Torch库的Python实现的深度学习库,目前被广泛应用于学术界和工业界,相较于Tensorflow2.x,PyTorch在API的设计上更加简洁、优雅和易懂。
2. 前置技术依赖
3. 环境配置
- 查看显卡 cuda 版本
|
|
PyTorch 向下兼容 CUDA 版本,根据 CUDA 版本安装合适的 PyTorch 版本: