如何让Jupyter Notebook支持pytorch
前言
因为最近想学深度学习,所以迷上了python,感觉Jupyter Notebook很便捷很方便,所以在自己的电脑上面搞一搞,过程很曲折,不过后来终于成功了,在此记录一下。
Anaconda安装
首先,安装了Anaconda3,有了Anaconda的环境就会很方便。Anaconda官网的下载巨慢无比,所以我采用了清华镜像下载:清华大学开源软件镜像站,进去后根据自己的需求选择匹配的版本,我选了最新版本:
接下来安装就可以了。
遇到VS Code那一步跳过就可以了。
完成以后开始菜单中就会出现Anaconda3,单击Anaconda Prompt会出现一个类似于cmd的命令行
这样就表明Anaconda已安装成功了。
此时单击Jupyter Notebook或者在命令行中输入Jupyter Notebook都会打开Jupyter Notebook,而且自动搭载好了python3的环境,普通的python代码已经可以运行了。
不过还没有pytorch库
那么接下来就是最折腾的,安装pytorch环节!
安装pytorch
Anaconda环境管理
进入Anaconda后默认是base环境,为了方便管理可以创建不同的子环境,他们之间相互分格开的。
conda create -n name python=你的版本
可以创建环境,其中名字是随便起的,python版本根据自己的定,我的是3.8。
先创建一个pytorch环境,遇到安装提示输入y,然后安装。conda info -e
可以查看当前conda中的环境,其中*号表示当前所处的环境。
conda activate name
可以进入新环境。
另外,conda activate
可以退出当前环境,conda remove -n name --all
可以删除不想要的环境。
pytorch安装
之前为了方便起见,创建了一个新环境pytorch,有关pytorch的都会安装在pytorch环境中。
进入pytorch官网:pytorch官网
根据自己的情况选择合适的版本,我也是根据网上的建议选了CUDA 9.2(支持gpu的),单独的cpu版本请选择None。
复制官网给出的命令行,直接进行安装就可以了,记得要在需要安装的环境中,根据提示输入y进行安装,接下来的等待可能会有点漫长。
安装完成之后,命令行输入python
,然后输入import torch
,如果没有报错就证明安装成啦。
不过别以为这样就可以用了,我刚开始就这么想的,事实证明我还是太年轻。
调用pytorch虚拟环境
之所以还不能用,我觉得是因为前面仅仅是安装上了,但是jupyter notebook还没有调用它。
- 首先,以管理员身份打开Anaconda Prompt,然后安装插件。
命令行输入conda install nb_conda
,遇到提示输入y安装。
安装倒是成了,不过中间有错误,我也不知道为啥,反正最后结果不影响,在这儿把它贴出来。
- 然后进入创建的pytorch环境,命令行输入
conda install ipykernel
,然后根据提示输入y进行安装。安装成功是这个样子的:
激动地打开Jupyter Notebook,发现语言多了两个:
对于使用了pytorch库的代码,选择自己创建的那个pytorch环境,因为里面是装过torch的
然后发现没毛病了,可以顺利运行了
后记
prtorch的安装及环境搭载网上也是众说纷纭,我这里根据我自己的情况作了一下总结,其实也是方便我以后自己学习的,感谢两位老哥的帖子:https://www.cnblogs.com/zhouzhiyao/p/11784055.html,https://blog.****.net/weixin_41899102/article/details/104123957
Jupyter Notebook确实很好用,不但可以直接运行代码,还可以直接进行文字记录、公式编写,对于小白来说真的是发现 了宝藏。而且Jupyter Notebook还可以支持很多种语言的,Jupyter Notebook支持语言列表,后续把我会的语言都搞上去,哇想想还有点小激动呢。