Torch使用Lua语言编写,2016年10月份,facebook基于Torch实现了python版本——PyTorch。
以下是目前常用的深度学习框架:
一、使用深度学习框架的好处
GPU加速
- 自动求导
- 提供了很多神经网络常用的API
二、开发环境安装
Python3 + Anaconda + CUDA
1、Anaconda安装
2、CUDA安装
(1)安装好后验证
安装好后,安装目录下的 nvcc.exe 可执行程序是CUDA的编译器
验证版本:
3、PyTorch安装
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch