如何在AWS SageMaker中修改软件包安装目录？

1. 确认目标磁盘挂载状态 #

先检查/dev/nvme1n1是否已挂载，以及具体挂载路径：

df -h

如果未挂载，先创建挂载点并完成挂载（示例挂载到/mnt/nvme1）：

sudo mkdir -p /mnt/nvme1
sudo mount /dev/nvme1n1 /mnt/nvme1

给当前用户赋予该目录的读写权限：

sudo chown -R $USER:$USER /mnt/nvme1

2. 针对pip工具指定安装路径 #

单次安装临时生效 #

安装包时直接通过--target参数指定路径：

pip install <你的包名> --target /mnt/nvme1/python-packages

安装后需要将该路径加入Python的搜索路径才能正常导入包：

import sys
sys.path.append('/mnt/nvme1/python-packages')

永久修改pip默认路径 #

创建pip配置文件，让后续所有pip安装都默认指向目标目录：

mkdir -p ~/.pip
echo "[global]
target = /mnt/nvme1/python-packages" > ~/.pip/pip.conf

同样需要把目标路径加入Python搜索路径，可将上述Python代码添加到Notebook开头，或者写入Notebook初始化脚本。

3. 针对conda环境指定安装路径 #

如果使用conda管理环境，直接在目标目录创建新环境：

conda create --prefix /mnt/nvme1/conda-env <你的包名>

激活该环境后，后续安装的包都会自动存入目标目录：

conda activate /mnt/nvme1/conda-env

4. 配置环境变量确保路径生效 #

将目标包路径加入PYTHONPATH，避免每次手动添加：

export PYTHONPATH="/mnt/nvme1/python-packages:$PYTHONPATH"

若要永久生效，将上述命令写入~/.bashrc或~/.profile文件。

验证安装结果 #

安装完成后，检查包的实际安装位置：

pip show <你的包名> | grep Location

若输出路径为/mnt/nvme1下的目录，则说明配置成功。

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