已下载PyTorch但VS Code无法运行指定单元格，导入命令执行超时

1. 确认Jupyter内核与PyTorch环境匹配 #

• 打开VS Code的内核选择器（右上角或通过Ctrl+Shift+P调用命令面板输入Select Kernel），确保选中的内核是你安装PyTorch的目标环境（比如Anaconda虚拟环境）。
• 先在终端激活对应环境（conda activate 你的环境名），执行python -c "import torch; print(torch.__version__)"，确认环境本身能正常导入PyTorch，再回到VS Code选择该内核。

2. 核对内核的Python路径 #

• 在Jupyter单元格中运行import sys; print(sys.executable)，查看当前内核使用的Python路径。
• 对比你安装PyTorch时的Python路径，若不一致，通过命令面板输入Jupyter: Specify Local Kernel，手动选择对应环境的Python可执行文件路径。

3. 清理缓存并重启VS Code #

• 关闭VS Code，删除以下缓存目录：
  • Windows：%APPDATA%\Code\Cache 和 %APPDATA%\Code\CachedData
  • macOS：~/Library/Caches/Code 和 ~/Library/Application Support/Code/CachedData
  • Linux：~/.cache/code 和 ~/.config/Code/CachedData
• 重启VS Code后重新打开Jupyter notebook测试。

4. 重新安装PyTorch（解决环境冲突） #

• 先卸载当前环境的PyTorch相关包：

  pip uninstall torch torchvision torchaudio -y
  

• 根据你的系统、CUDA版本，从PyTorch官网获取对应安装命令，比如CUDA 11.8的安装命令：

  pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
  

• 安装完成后在终端测试导入，再回到VS Code运行单元格。

5. 排查扩展冲突 #

• 暂时禁用VS Code中除Jupyter、Python外的其他扩展，重启VS Code后测试PyTorch导入，排除扩展干扰。

内容的提问来源于stack exchange，提问作者Mani Kanta Lokesh Buragadda