1:深度学习和pytorch(深度学习的常用框架,开发环境的安装)

1:常用的深度深度学习框架

  1. Theano
  2. tensorflow
  3. keras
  4. MXNet
  5. CNTK
  6. Caffe
  7. 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安装成功。

原文地址:https://www.cnblogs.com/jiafeng1996/p/14979244.html