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

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/
在这里插入图片描述
下滑找到如下界面,即可选择安装方式:

选择合适并版本复制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
 安装好后进行测试:

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

 
原文地址:https://www.cnblogs.com/cy0628/p/14052014.html