在Linux系统中安装深度学习框架Pytorch
01
PyTorch和TensorFlow
PyTorch:一个开源的Python机器学习库 , 基于Torch , 用于自然语言处理等程序 。 使用Python作为开发语言 , 是一个以Python优先的深度学习框架 。 本质上是Numpy的替代者 。
TensorFlow:Tensor(张量)意味着N维数组 , Flow(流)意味着基于数据流图的计算 , TensorFlow为张量从流图的一端流动到另一端计算过程 。
TensorFlow是将复杂的数据结构传输至人工智能神经网中进行分析和处理过程的系统 , 可以看成是一个嵌入Python的编程语言 。
从目前的文档看 , TensorFlow支持CNN、RNN和LSTM算法 , 这都是目前在Image , Speech和NLP最流行的深度神经网络模型 。
PyTorch更有利于研究人员、爱好者、小规模项目等快速搞出原型 。 而TensorFlow更适合大规模部署 , 特别是需要跨平台和嵌入式部署时 。
02
在Linux下载Pytorch
由推文:在Linux系统中安装anaconda的详细实操步骤 可知 , 安装PyTorch可以通过anaconda进行安装 。
详细步骤:
1. 在确保anaconda安装成功后 , 进入下一步 。
在上一篇推文最后中 , 打开新的终端验证anaconda安装成功时 , 会发现命令前面有(base) 。
2.为PyTorch创建虚拟环境 。
为pytorch创建一个虚拟环境 , 这是推荐的做法 , 这样在创建其他环境如tensorflow时 , 就不会互相冲突 。 在终端输入:conda create -n pytorch python=3.7
文章插图
如果安装的是Anaconda2 , 安装后会出现:
文章插图
【在Linux系统中安装深度学习框架Pytorch】再输入 conda activate pytorch进入该环境 , 此外conda deactivate退出该环境 。 进入该环境后 , 命令行前面有(pytorch) 。
3.添加清华源 。
否则会下载过慢导致失败 。 在终端输入以下内容 , 可直接粘贴复制 。 在Linux系统终端复制时按Ctrl+Shift+v 。
conda config --add channels
conda config --add channels
conda config --add channels
conda config --add channels
conda config --add channels
conda config --add channels
conda config --add channels
conda config --set show_channel_urls yes
最后一句是设置安装路径可见 。 可输入cat ~/.condarc查看是否成功 。
补充:.condarc文件在Home文件中 , 是隐藏文件 , 可按Ctrl+H将其显示 。
文章插图
打开后可以看到:
文章插图
4打开PyTorch官网 。
官网:#anaconda , 根据提示进行选择:
文章插图
在Linux终端输入指令 , 注意官网根据你的选择给出的指令是:
conda install pytorch torchvision cpuonly -c pytorch
而在终端输入的指令时将后面的 -c pytorch 删除 , 这样下载的路径才是清华源 。
即输入:
conda install pytorch torchvision cpuonly
输入后等待 , 直到安装完成 。
5.检测是否安装成功
打开新的终端 , 输入python , 再输入import torch , 没有报错;再输入import torchvision , 也没有报错 , 如图所示 。 此时即安装成功 。
文章插图
- vivo一款新机现身跑分网!运存和系统信息通通曝光
- 短短几个月,安装包从200M涨到354M,微信越来越臃肿了?
- 人瑞人才(06919):未来3年系统平台将发力智能化,打造职业生态链平台
- 便捷安装、功能强大的手机稳定器 大疆OM 4
- 消费者报告 | 美团充电宝电量不足也扣费,是质量问题还是系统缺陷?
- 谷歌修复Pixel 5系统音量问题 快门音效不再吵
- Axiomtek推出AIE100-903-FL-NX AI Edge系统 可在恶劣环境下运行
- 官宣!11款vivo系机型开始公测最新系统OriginOS
- 鸿蒙OS大一统需产业合力
- Linux Kernel 5.10.5发布:禁用FBCON加速滚动特性