[ 机器学习 ] 关于Jupyter Notebook中pytorch模块import失败的问题

0x01、问题描述

在使用WSL搭建Jupyter进行代码测试的时候

发现Miniconda（虚拟环境均适用）中安装的pytorch在Jupyter里面import失败

但在python解释器的命令模式里可以测试import成功

并且torch.cuda_available()打印True

以前用的是IDEA没怎么用Jupyter，搜索经验贴国内的答主大多都在重装，测试无效

0x02、解决流程

大致要先对虚拟环境的概念有个直观了解，再来看解决思路：

首先Jupyter Notebook要确保IPython Kernel是可用的

而我们必须手动添加一个具有不同版本Python的内核或虚拟环境

确保环境已经用conda activate xxx激活之后，安装ipykernel（为Jupyter提供IPython内核）

pip install --user ipykernel

接下来，添加虚拟环境到Jupyter

python -m ipykernel install --user --name=your_env_name

如果打印出以下结果：

Installed kernelspec your_env_name in /xxx/xxx...

cd打开上述路径（Debian/Ubuntu），ls查看kernel.json

vim打开check一下没问题就OK

{

 "argv": [

  "/xxx/xxx...",

  "-m",

  "ipykernel_launcher",

  "-f",

  "{connection_file}"

 ],

 "display_name": "your_env_name",

 "language": "python"

 ...  # 这里会可能有一些不一样的信息，但问题不大

}

0x03、测试结果

启动Jupyter Notebook并在Kernel–change kernel中选择安装好torch的环境

连接成功后进行测试，问题解决！

测试结果

相关链接：

https://janakiev.com/blog/jupyter-virtual-envs/