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

2023-03-02 13:53:29 浏览数 (2)

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内核)

代码语言:txt复制
pip install --user ipykernel

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

代码语言:txt复制
python -m ipykernel install --user --name=your_env_name

如果打印出以下结果:

代码语言:txt复制
Installed kernelspec your_env_name in /xxx/xxx...

cd打开上述路径(Debian/Ubuntu),ls查看kernel.json

vim打开check一下没问题就OK

代码语言:txt复制
{

 "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/

问题如果未解决请评论区留言,或对照以上链接检查,可以去Github Issue中找同类型问题......

0 人点赞