如何查找PyTorch源码文件位置以进行编辑（含torch.utils.tensorboard示例）

1. 定位PyTorch模块的源码位置 #

要找到torch.utils.tensorboard这类模块的源码文件，最直接的方法是在Python环境中打印模块的__file__属性：

• 打开Python终端或在脚本中执行：

import torch.utils.tensorboard
print(torch.utils.tensorboard.__file__)

执行后会输出该模块的具体文件路径（例如/your/python/env/site-packages/torch/utils/tensorboard/__init__.py），对应的源码文件就位于这个路径的目录下。

如果需要找到整个PyTorch的主安装目录，同样可以通过torch核心模块的__file__属性定位：

import torch
print(torch.__file__)

输出路径的上一级目录就是PyTorch的主目录，包含所有核心模块的源码文件。

2. PyTorch相关包的下载路径 #

不同安装方式对应的下载路径不同：

• pip安装：下载的缓存包默认存储在：
  • Linux/macOS：~/.cache/pip/downloads
  • Windows：%APPDATA%\pip\Cache\downloads
• conda安装：下载的包文件存放在conda的pkgs目录，例如：
  • Linux/macOS：~/miniconda3/pkgs（根据你的conda实际安装路径调整）
  • Windows：C:\Users\<你的用户名>\miniconda3\pkgs
• 源码编译安装：下载的源码包位置取决于你下载时指定的目录，编译后的输出文件则由编译时的配置参数决定。

注意事项 #

手动编辑PyTorch源码后，需要重启Python环境才能生效；如果后续通过包管理工具更新PyTorch，你修改的源码会被覆盖，建议通过fork官方仓库、自定义编译的方式来持久化修改。

内容的提问来源于stack exchange，提问作者Jinwoo Mo