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中找同类型问题......