大帅学习笔记——pytorch入门（1）pytorch安装

最新推荐文章于 2024-09-02 14:43:35 发布

肖大帅呀

最新推荐文章于 2024-09-02 14:43:35 发布

阅读量275

点赞数 1

分类专栏：大帅学习笔记文章标签：深度学习 python pytorch

本文链接：https://blog.csdn.net/weixin_43509207/article/details/107416325

版权

大帅学习笔记专栏收录该内容

3 篇文章 0 订阅

订阅专栏

大帅出品，必属精品

记录学习pytorch的全过程，入门教程使用deeplizard的pytorch教程网站里包括视频和讲解博文，因为视频在YouTube上，这里贴出B站上的中文字幕课程视频。新人上路，希望大家多多指正。

1. 为什么是pytorch

在深度学习领域，框架已经比较成熟，再tensorFlow2.0发布之后，工业界基本是tensorFlow一家独大（caffe哭)，这很大程度上是因为tensorFlow庞大社区的支持和设备管理上简单的操作。而pytorch因为与python是好基友而且和numpy等科学计算库相容性较好，且容易调试，成了科研界的爱宠。由于作者是为了做科研项目（捡垃圾吃）而学习所以使用torch。这里也推荐要读研的小伙伴可以首选学习torch。

2. pytorch安装

首先，作为计算机专业的学生，安装东西当然要去官网啦
那么->torch传送门
本机使用anaconda的python3.7版本为准，且小破本没有N卡。。。

假设你已经安装anaconda（或者没有），那么打开传送门，选择合适你的torch类型 pytorch安装

以上

Package中如果是anaconda安装的python就选择Conda，否则选Pip
CUDA显卡号，需要对应自己电脑的版本。我的笔记本没有独显所以选None

复制Run this Command中的命令，打开conda的命令行，CV这条命令（如果是pip，那就打开自己电脑的终端CV对应的命令）。在这里插入图片描述

装的时候。。。网不是很好，该怎么做大家是知道的。

这里就不展示安装的具体过程了，反正只要y个不停就完事了呗。安装结束之后你的电脑上就已经有了pytorch和对应版本的torchvision（里面会有数据集和一些别的功能）。

3. 验证安装

验证安装需要先导入这两个包

import torch
import torchvision

再用.__version__方法查看版本

print(torchvision.__version__)
print(torch.__version__)

如果没有猜错的话，它应该是这样的
在这里插入图片描述
这里基本不会有什么问题，唯一我能想到的就是torch版本和torchvision版本不匹配问题，解决方法：卸载旧版本的torch或者torchvision，重新安装。
如果是用conda的同学可能习惯在conda的包管理器里直接找包下载，当然懒的时候这确实很好用，但是！！！torch不行版本差着十万八千里。

有能（金）力（钱）的同学如果使用N卡的话还需要验证你的显卡是否已经和torch完美结合。cuda的使用也很简单，只要在所有的计算代码后加上.cuda()即可，因为本机上全部使用cou计算，所以今后所有的代码都是默认不用cuda的，有条件的小伙伴记得加哦。
在这里插入图片描述
简单解释一下为什么需要用到显卡计算。因为显卡有着巨大量的计算核心，这是你的cpu所不具备的。举个简单的例子(因为实在没看懂大佬的讲解）：他们的区别就像1个大学生和1000个小学生一样，当面对高数的时候大学生是主力，但是当计算冗杂且并不复杂的神经网络计算时，就像做一千道两位数加减法。大学生再怎么厉害也需要很久很久，可是一千个小学生只需要20秒。