You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

PyCharm 2025.1 Jupyter内核导入torch报libnccl.so.2缺失,终端环境正常

解决PyCharm Jupyter中ImportError: libnccl.so.2找不到的问题

针对你遇到的问题——同一个虚拟环境下,终端ipython能正常导入torch(2.2.2+cu121),但PyCharm 2025.1的Jupyter Notebook里却报找不到libnccl.so.2,且LD_LIBRARY_PATH路径一致的情况,给你几个实操的解决方向:

  • 确认Jupyter内核与虚拟环境绑定正确
    有时候PyCharm会默认使用全局Jupyter内核,即便你选了虚拟环境也可能出现内核不匹配的情况。手动在虚拟环境中重新注册内核:

    # 先激活你的虚拟环境
    source /path/to/your/env/bin/activate
    # 安装ipykernel到当前环境
    pip install ipykernel
    # 注册内核,替换成你的环境名称
    python -m ipykernel install --user --name=your-env-name
    

    然后在PyCharm的Jupyter Notebook中,切换到刚注册的内核再测试导入。

  • 在Jupyter中临时指定nccl库路径
    虽然LD_LIBRARY_PATH一致,但Jupyter的环境变量加载逻辑可能和终端有差异。先找到你的虚拟环境中torch的lib目录(比如/path/to/your/env/lib/python3.10/site-packages/torch/lib),在Jupyter的代码开头添加:

    import os
    # 替换成你实际的torch lib路径
    os.environ['LD_LIBRARY_PATH'] += ':/path/to/your/env/lib/python3.x/site-packages/torch/lib'
    import torch
    

    这样强制让Jupyter加载torch自带的nccl库。

  • 检查PyCharm运行配置的环境变量覆盖
    打开PyCharm的「Run Configuration」,找到Jupyter Notebook的配置项,查看「Environment variables」里是否单独设置了LD_LIBRARY_PATH,导致覆盖了系统的路径。如果有,删除自定义的设置,保持和终端一致。

  • 重新安装torch确保nccl库完整
    进入虚拟环境,重新安装对应版本的torch,确保nccl相关库被正确下载:

    pip uninstall torch torchvision torchaudio
    pip install torch==2.2.2+cu121 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
    

    安装完成后,去虚拟环境的torch/lib目录下确认libnccl.so.2是否存在。

内容的提问来源于stack exchange,提问作者Victor Zuanazzi

火山引擎 最新活动