如何指定pip安装torch的下载目录？解决磁盘空间不足问题

你遇到的问题是混淆了pip的「安装目录」和「下载缓存目录」：-t参数是指定包最终安装到哪个目录，但pip下载的安装包缓存默认存在根盘的~/.cache/pip目录里，这才是占用你根盘空间的核心原因。以下是几种直接修改下载缓存路径的实用方法：

1. 单次命令临时指定缓存目录 #

直接在pip install命令后添加--cache-dir参数，指定大磁盘上的目录作为缓存路径：

"/media/.../python3" -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118 -t "/directory/" --cache-dir "/media/your-large-disk/pip-cache/"

替换/media/your-large-disk/pip-cache/为你实际的大磁盘空目录即可，这样几GB的安装包会直接存在这个目录，不会占用根盘空间。

2. 永久修改pip缓存路径 #

如果不想每次都手动加参数，可以配置pip默认使用大磁盘缓存：

方法一：修改pip配置文件 #

在Debian系统中，创建或编辑用户目录下的pip配置文件：

mkdir -p ~/.config/pip
nano ~/.config/pip/pip.conf

在文件中添加以下内容（替换为你的大磁盘目录）：

[global]
cache-dir = /media/your-large-disk/pip-cache/

保存退出后，后续所有pip命令都会自动使用这个缓存目录。

方法二：设置环境变量 #

• 临时生效：在终端先执行以下命令，再运行pip install：

  export PIP_CACHE_DIR="/media/your-large-disk/pip-cache/"
  

• 永久生效：编辑你的shell配置文件（比如~/.bashrc或~/.zshrc），添加上面的export语句，然后执行source ~/.bashrc（对应你的rc文件）使配置生效。

3. 符号链接替代方案 #

如果你不想修改pip配置，也可以把默认的缓存目录软链到大磁盘：

1. 先备份并删除原缓存目录（如果不需要保留原缓存可跳过备份）：

   mv ~/.cache/pip /media/your-large-disk/pip-cache-backup
   rm -rf ~/.cache/pip
   

2. 创建软链指向大磁盘目录：

   ln -s /media/your-large-disk/pip-cache/ ~/.cache/pip
   

这样pip依然会往~/.cache/pip写入数据，但实际存储位置是你指定的大磁盘目录。

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