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

2024-01-10 15:28:07

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/

文章来源:https://blog.csdn.net/2301_81940605/article/details/135473367
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。