当前位置 : 主页 > 编程语言 > python >

ubuntu20.04安装pytorch教程及其注意事项

来源:互联网 收集:自由互联 发布时间:2022-06-15
Pytorch介绍 PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神

Pytorch介绍

PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。PyTorch既可以看作加入了GPU支持的numpy,同时也可以看成一个拥有自动求导功能的强大的深度神经网络。

它具有以下优势:
PyTorch是相当简洁且高效快速的框架
设计追求最少的封装
设计符合人类思维,它让用户尽可能地专注于实现自己的想法
与google的Tensorflow类似,FAIR的支持足以确保PyTorch获得持续的开发更新
PyTorch作者亲自维护的论坛 供用户交流和求教问题
入门简单

至于其更广泛更具体的优势,参照知乎的讨论:​​https://www.zhihu.com/question/65578911/answer/565574377​​

Pytorch安装

pytorch的安装难度要远低于tensorflow。百度搜索并进入​​https://pytorch.org/get-started/locally/​​​ubuntu20.04安装pytorch教程及其注意事项_机器学习
下滑找到如下界面,即可选择安装方式:
ubuntu20.04安装pytorch教程及其注意事项_tensorflow_02

选择合适并版本复制Run this Command:即可进行安装(不推荐使用方法,因为下载速度很慢)。

注意事项:安装过程可能比较缓慢,可以用以下命令(推荐)

pip --default-timeout=1000 install --index-url https://mirrors.aliyun.com/pypi/simple torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

 安装好后进行测试:

ubuntu20.04安装pytorch教程及其注意事项_tensorflow_03

 输入以上命令即可完成测试是否成功安装pytorch以及pytorch版本。

 

网友评论