如何让Jupyter Notebook支持pytorch

前言

因为最近想学深度学习,所以迷上了python,感觉Jupyter Notebook很便捷很方便,所以在自己的电脑上面搞一搞,过程很曲折,不过后来终于成功了,在此记录一下。

Anaconda安装

首先,安装了Anaconda3,有了Anaconda的环境就会很方便。Anaconda官网的下载巨慢无比,所以我采用了清华镜像下载:清华大学开源软件镜像站,进去后根据自己的需求选择匹配的版本,我选了最新版本:
如何让Jupyter Notebook支持pytorch_第1张图片
接下来安装就可以了。
如何让Jupyter Notebook支持pytorch_第2张图片
如何让Jupyter Notebook支持pytorch_第3张图片
遇到VS Code那一步跳过就可以了。
完成以后开始菜单中就会出现Anaconda3,单击Anaconda Prompt会出现一个类似于cmd的命令行
如何让Jupyter Notebook支持pytorch_第4张图片
如何让Jupyter Notebook支持pytorch_第5张图片
这样就表明Anaconda已安装成功了。

此时单击Jupyter Notebook或者在命令行中输入Jupyter Notebook都会打开Jupyter Notebook,而且自动搭载好了python3的环境,普通的python代码已经可以运行了。
如何让Jupyter Notebook支持pytorch_第6张图片
在这里插入图片描述
不过还没有pytorch库如何让Jupyter Notebook支持pytorch_第7张图片
那么接下来就是最折腾的,安装pytorch环节!

安装pytorch

Anaconda环境管理

进入Anaconda后默认是base环境,为了方便管理可以创建不同的子环境,他们之间相互分格开的。

conda create -n name python=你的版本可以创建环境,其中名字是随便起的,python版本根据自己的定,我的是3.8。

先创建一个pytorch环境,遇到安装提示输入y,然后安装。
conda info -e可以查看当前conda中的环境,其中*号表示当前所处的环境。
如何让Jupyter Notebook支持pytorch_第8张图片

conda activate name可以进入新环境。
如何让Jupyter Notebook支持pytorch_第9张图片

另外,conda activate可以退出当前环境,conda remove -n name --all可以删除不想要的环境。

pytorch安装

之前为了方便起见,创建了一个新环境pytorch,有关pytorch的都会安装在pytorch环境中。

进入pytorch官网:pytorch官网

根据自己的情况选择合适的版本,我也是根据网上的建议选了CUDA 9.2(支持gpu的),单独的cpu版本请选择None。
如何让Jupyter Notebook支持pytorch_第10张图片

复制官网给出的命令行,直接进行安装就可以了,记得要在需要安装的环境中,根据提示输入y进行安装,接下来的等待可能会有点漫长。
如何让Jupyter Notebook支持pytorch_第11张图片

安装完成之后,命令行输入python,然后输入import torch,如果没有报错就证明安装成啦。
在这里插入图片描述

安装mxnet

后来发现还要安装mxnet库,后来找啊找,发现一个链接:《动手学习深度学习》环境搭建详细教程,里面有关各种库的安装讲的非常详细直观,套用里面的方法一次成功,在这记录下来。

调用pytorch虚拟环境

不过别以为这样就可以用了,我刚开始就这么想的,事实证明我还是太年轻。
如何让Jupyter Notebook支持pytorch_第12张图片

之所以还不能用,我觉得是因为前面仅仅是安装上了,但是jupyter notebook还没有调用它。

  • 首先,以管理员身份打开Anaconda Prompt,然后安装插件。

命令行输入conda install nb_conda,遇到提示输入y安装。
安装倒是成了,不过中间有错误,我也不知道为啥,反正最后结果不影响,在这儿把它贴出来。
如何让Jupyter Notebook支持pytorch_第13张图片

  • 然后进入创建的pytorch环境,命令行输入conda install ipykernel,然后根据提示输入y进行安装。安装成功是这个样子的:
    如何让Jupyter Notebook支持pytorch_第14张图片

激动地打开Jupyter Notebook,发现语言多了两个:
如何让Jupyter Notebook支持pytorch_第15张图片

对于使用了pytorch库的代码,选择自己创建的那个pytorch环境,因为里面是装过torch的
如何让Jupyter Notebook支持pytorch_第16张图片

然后发现没毛病了,可以顺利运行了
如何让Jupyter Notebook支持pytorch_第17张图片

后记

prtorch的安装及环境搭载网上也是众说纷纭,我这里根据我自己的情况作了一下总结,其实也是方便我以后自己学习的,感谢两位老哥的帖子:https://www.cnblogs.com/zhouzhiyao/p/11784055.html,https://blog.csdn.net/weixin_41899102/article/details/104123957

Jupyter Notebook确实很好用,不但可以直接运行代码,还可以直接进行文字记录、公式编写,对于小白来说真的是发现 了宝藏。而且Jupyter Notebook还可以支持很多种语言的,Jupyter Notebook支持语言列表,后续把我会的语言都搞上去,哇想想还有点小激动呢。

你可能感兴趣的:(如何让Jupyter Notebook支持pytorch)