最近在学习深度学习的过程当中需要用到pytorch环境,并部署到jupyter notebook当中使用。在这过程当中遇到许多坑,特此记录下来,希望可以帮助到其他小伙伴。
注意:本文基于电脑上已经安装好了Anaconda
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环境中安装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就可以成功操作了。
由此,问题解决!