1:常用的深度深度学习框架
- Theano
- tensorflow
- keras
- MXNet
- CNTK
- Caffe
- pytorch:优点:GPU加速,自动求导,常用API
2:开发环境的安装
windows下步骤:
1:Anaconda安装:安装python以及其集成工具,由于anaconda带有python编辑器,因此只需安装Anaconda就行。
进入Anaconda官网(点击windows->python3.7下载)->双击exe文件
[注]:注意勾选第一个选项
[注]:由于是轻量级来开发,右边可以跳过
安装成功之后进入cmd:
conda list可以查看已经安装了哪些包
conda --version查看版本
2:cuda的安装
[注]cuda必须运行在nvidia显卡上
进入cuda官网(选择windows版本以及cuda版本如下图)->双击exe文件:
安装成功之后:进入cmd
输入:nvcc -V
显示cuda版本信息则安装成功
【注】如果安装出现如下错误
解决办法:
打开计算机的高级设置:将该路径添加到系统变量中
3:pytorch安装
进入pytorch官网->选择如下:
以管理员身份运行cmd:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch(可以去掉-c pytorch加快安装速度)
可以直接使用指定版本的命令行:
参考:https://pytorch.org/get-started/previous-versions/
【注:】在所有流程请注意python,cuda,torch版本的匹配
版本匹配参考博客:https://pytorch.org/get-started/previous-versions/
【注】如何检验torch是否安装成功:
进入cmd输入:python
import torch
print(torch.cuda.is_available())#检查cuda是否可用
print(torch.__version__)#查看torch的版本
如果命令可以运行成功则表明torch安装成功。