Python实现类库安装指定目录的示例详解

  TARGET_DIR="/gemine/code/python311"


wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.0.post1/flash_attn-2.7.0.post1+cu12torch2.4cxx11abiFALSE-cp311-cp311-linux_x86_64.whl

pip install flash_attn-2.6.3+cu118torch2.4cxx11abiFALSE-cp311-cp311-linux_x86_64.whl \
    --target=$TARGET_DIR \
    --no-deps \
    --no-build-isolation

import sys
sys.path.insert(0, '/gemine/code/python311')

# 现在可以导入该目录下的包
import flash_attn

# 编辑 bashrc
echo 'export PYTHONPATH="/gemine/code/python311:$PYTHONPATH"' >> ~/.bashrc

# 立即生效
source ~/.bashrc

# 找到 Python 站点包目录
python -m site --user-site

# 创建 .pth 文件（假设输出是 /home/user/.local/lib/python3.11/site-packages）
echo "/gemine/code/python311" > /home/user/.local/lib/python3.11/site-packages/custom_path.pth

# 找到当前环境的 site-packages
python -c "import site; print(site.getsitepackages()[0])"

# 创建 .pth 文件
echo "/gemine/code/python311" > $(python -c "import site; print(site.getsitepackages()[0])")/custom_lib.pth

import sys
import os

# 自动添加自定义库路径
TARGET_DIR = '/gemine/code/python311'

if TARGET_DIR not in sys.path:
    sys.path.insert(0, TARGET_DIR)
    print(f"已添加 {TARGET_DIR} 到 Python 路径")

# 验证
try:
    import flash_attn
    print(f"flash-attn 加载成功，版本: {flash_attn.__version__}")
except ImportError as e:
    print(f"加载失败: {e}")

# 一键添加（通用方法）
python << 'EOF'
import site
import os

target_dir = "/gemine/code/python311"
site_packages = site.getsitepackages()[0]
pth_file = os.path.join(site_packages, "custom_lib.pth")

# 写入 .pth 文件
with open(pth_file, 'w') as f:
    f.write(target_dir + '\n')

print(f"已创建: {pth_file}")
print(f"内容: {target_dir}")
EOF

# 检查 Python 路径
python -c "import sys; print('/gemine/code/python311' in sys.path)"

# 检查能否导入 flash-attn
python -c "import flash_attn; print(flash_attn.__version__)"