Ubuntu系统中PyTorch的安装路径主要取决于安装方式(pip/conda)和是否使用虚拟环境,以下是常见场景的具体路径说明:
dist-packages目录,路径格式为/usr/local/lib/pythonX.X/dist-packages/torch(X.X为Python主版本号,如3.8、3.10)。例如,Ubuntu系统默认Python 3.8时,路径为/usr/local/lib/python3.8/dist-packages/torch。--user参数):PyTorch会安装到用户目录下的~/.local/lib/pythonX.X/site-packages/torch。site-packages目录,路径格式为~/anaconda3/envs/环境名/lib/pythonX.X/site-packages/torch(环境名为用户创建的虚拟环境名称,如pytorch_env;X.X为Python版本)。例如,名为pytorch的虚拟环境中,Python 3.6的路径为/home/用户名/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch。site-packages一致,如~/anaconda3/lib/python3.X/site-packages/torch。无论采用哪种安装方式,均可通过以下Python命令快速确认PyTorch的实际安装路径:
import torch
print(torch.__path__)
该命令会输出PyTorch模块的安装路径列表(通常只有一个路径)。
--prefix参数指定),路径会跟随自定义设置,但此场景较少见。