温馨提示×

Ubuntu系统中PyTorch的安装路径

小樊
52
2025-10-06 06:38:31
栏目: 智能运维

Ubuntu系统中PyTorch的安装路径主要取决于安装方式(pip/conda)和是否使用虚拟环境,以下是常见场景的具体路径说明:

1. pip安装(系统全局或用户级)

  • 系统全局安装(无虚拟环境):PyTorch会被安装到Python的dist-packages目录,路径格式为/usr/local/lib/pythonX.X/dist-packages/torchX.X为Python主版本号,如3.8、3.10)。例如,Ubuntu系统默认Python 3.8时,路径为/usr/local/lib/python3.8/dist-packages/torch
  • 用户级pip安装--user参数):PyTorch会安装到用户目录下的~/.local/lib/pythonX.X/site-packages/torch

2. conda安装(虚拟环境)

  • conda虚拟环境:PyTorch会被安装到对应虚拟环境的site-packages目录,路径格式为~/anaconda3/envs/环境名/lib/pythonX.X/site-packages/torch环境名为用户创建的虚拟环境名称,如pytorch_envX.X为Python版本)。例如,名为pytorch的虚拟环境中,Python 3.6的路径为/home/用户名/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch
  • conda默认环境(未创建虚拟环境时):路径与系统Python的site-packages一致,如~/anaconda3/lib/python3.X/site-packages/torch

3. 验证安装路径的方法

无论采用哪种安装方式,均可通过以下Python命令快速确认PyTorch的实际安装路径:

import torch
print(torch.__path__)

该命令会输出PyTorch模块的安装路径列表(通常只有一个路径)。

注意事项

  • 若使用自定义安装路径(如通过--prefix参数指定),路径会跟随自定义设置,但此场景较少见。
  • 虚拟环境是隔离PyTorch安装的最佳实践,可避免不同项目间的依赖冲突。

0