Featured image of post PyTorch 学习笔记

PyTorch 学习笔记

学习记录 PyTorch,记录开发学习过程

PyTorch 实操笔记

1. 简单介绍

  • PyTorch 是一个深度学习框架,主要用于研究和生产环境
  • 由Meta AI(Facebook)人工智能研究小组开发的一种基于Lua编写的Torch库的Python实现的深度学习库,目前被广泛应用于学术界和工业界,相较于Tensorflow2.x,PyTorch在API的设计上更加简洁、优雅和易懂。

2. 前置技术依赖


3. 环境配置

  1. 查看显卡 cuda 版本
1
2
# windows 命令行中输入命令查看显卡信息
nvidia-smi

PyTorch 向下兼容 CUDA 版本,根据 CUDA 版本安装合适的 PyTorch 版本:


4. 基础用法


5. 常见问题与解决办法


6. tips


7. 参考资料


潇洒人间一键仙
Built with Hugo
Theme Stack designed by Jimmy