pytorch部署到jupyter notebook当中使用

前言:

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

安装pytorch

  1. 打开菜单栏Anaconda3(64-bit)文件夹当中的anaconda prompt控制台
    pytorch部署到jupyter notebook当中使用
    进入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环境:

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

; 解决办法:

一、重新在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

pytorch部署到jupyter notebook当中使用
此时再打开jupyter notebook可以发现new选项中多了关于pytorch的环境
pytorch部署到jupyter notebook当中使用
并且,菜单栏Anaconda3(64-bit)文件夹当中多了jupyter notebook(pytorch)
此时我们再调用pytorch就可以成功操作了。
pytorch部署到jupyter notebook当中使用
由此,问题解决!

Original: https://blog.csdn.net/weixin_45775136/article/details/125800300
Author: 会写bug的程序媛
Title: pytorch部署到jupyter notebook当中使用

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/707748/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球