如何让Jupyter Notebook支持pytorch

前言

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

Anaconda安装

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

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

安装pytorch

Anaconda环境管理

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

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

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

conda activate name可以进入新环境。
如何让Jupyter Notebook支持pytorch

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

pytorch安装

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

进入pytorch官网:pytorch官网

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

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

安装完成之后,命令行输入python,然后输入import torch,如果没有报错就证明安装成啦。
如何让Jupyter Notebook支持pytorch
不过别以为这样就可以用了,我刚开始就这么想的,事实证明我还是太年轻。
如何让Jupyter Notebook支持pytorch

调用pytorch虚拟环境

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

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

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

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

激动地打开Jupyter Notebook,发现语言多了两个:
如何让Jupyter Notebook支持pytorch

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

然后发现没毛病了,可以顺利运行了
如何让Jupyter Notebook支持pytorch

后记

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

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