根本原因是因为在jupyter notebook中默认是环境里面是不含torch模块的,所以必须创建一个新的环境,
所以如果我们要在jupyter notebook做一些特定框架下如pytorch的开发,我们就得在有pytorch的虚拟环境中安装jupyter的kernel,然后在jupyter中就可以切换kernel也就是虚拟开发环境了。
具体步骤:
第一步:创建一个新的虚拟环境,这里我电脑已经有了一个装有torch的环境pytorch(具体看你自己创建的torch环境叫什么名字,先打开anaconda prompt然后env list看自己的环境,我的叫pytorch),
为了不污染这个环境,我直接复制pytorch环境中的包到环境pytorch2中:conda create -n pytorch2 --clone
第二步:在虚拟环境下创建kernel:conda install -n pytorch2 ipykernel
第三步:激活虚拟环境:activate pytorch2
第四步:将该虚拟环境写进notebook的kernel中:python -m ipykernel install --user --name pytorch2 --display-name "python deep_pytorch"
这时你在base环境中输入jupyter notebook打开notebook,点击右上角的"new",这时“notebook:”列表下便会显示两个kerbel名称了。(这里要注意只能在base环境中打开jupyter notebook,因为只有base环境中装了它,虚拟环境中并没有装,而只是装了它的kernel)