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/