pytorch部署到jupyter notebook当中使用

前言:

最近在学习深度学习的过程当中需要用到pytorch环境,并部署到jupyter notebook当中使用。在这过程当中遇到许多坑,特此记录下来,希望可以帮助到其他小伙伴。
注意:本文基于电脑上已经安装好了Anaconda

安装pytorch

  1. 打开菜单栏Anaconda3(64-bit)文件夹当中的anaconda prompt控制台

    进入prompt后,默认是进入你的base环境。这里在安装pytorch时,为了方便后续的管理,新建一个pytorch虚拟环境,用来存放pytorch有关的环境。
    输入

conda create -n 虚拟空间名称 python==版本号

虚拟空间名称可以自定义,这里可定义为“pytorch”,python版本号要根据你自己电脑的python版本确定。我的创建命令是:

conda create -n pytorch python==3.7

输入conda info -e 可以查看当前conda中的环境。若创建成功,prompt中显示刚刚创建的pytorch环境:

在安装anaconda的同时,会将jupyter notebook一同安装,不过这里的jupyter notebook是base环境中的,而不是pytorch环境中的,这也是我一开始在虚拟环境当中安装了pytorh后一直
无法在jupyter notebook中调用的原因。
base中的jupyter:在base这里插入图片描述
要解决这个问题,这里有两种办法。

解决办法:

一、重新在base环境中安装pytorch;
二、在pytorch环境中安装jupyter notebook。

对于第一种方案,在安装时直接在base中配置即可,无需创建虚拟环境。本文主要介绍第二种方案。

首先输入conda activate pytorch激活当前虚拟环境

conda activate pytorch

输入以下指令,下载jupyter notebook

conda install nb_conda_kernels

完成后打开jupyter notebook

jupyter notebook

注:在输入该命令是,可能会弹出以下提示,点击确定即可,不影响后续使用聚jupyter notebook

此时再打开jupyter notebook可以发现new选项中多了关于pytorch的环境

并且,菜单栏Anaconda3(64-bit)文件夹当中多了jupyter notebook(pytorch)
此时我们再调用pytorch就可以成功操作了。

由此,问题解决!

物联沃分享整理
物联沃-IOTWORD物联网 » pytorch部署到jupyter notebook当中使用

发表评论