创建环境

# 如果是用 Miniconda 其中 env_name 是环境名字 python 后面是指定的 python 版本
conda create -n env_name python=3.xx.xx

# 如果不想或不能使用 conda 可以使用 venv
# 这个代码会在指定的路径下创建环境 env_name 也是环境名字
python -m venv /path/env_name

# 需要通过其他命令激活 然后就可以使用 pip 安装
/path/env_name/bin/activate

常用公共环境

Torch 环境

https://pytorch.org/get-started/locally/

# 测试版 pytorch 下载命令
pip3 install --pre torch torchvision torchaudio --index-url <https://download.pytorch.org/whl/nightly/cu126>

# 清华源 xx 表示你需要安装的库
pip install xx -i [<https://pypi.tuna.tsinghua.edu.cn/simple>](<https://pypi.tuna.tsinghua.edu.cn/simple>)

pip install albumentations opencv-contrib-python jupyter matplotlib lightning wandb einops pandas scikit-learn datasets transformers kornia bitsandbytes peft timm openvino nncf lightgbm triton torchmetrics ultralytics  heavyball -i [<https://pypi.tuna.tsinghua.edu.cn/simple>](<https://pypi.tuna.tsinghua.edu.cn/simple>)

#!/bin/bash

# 脚本默认使用 python3，如果需要指定版本，请先用 pyenv 激活
# pyenv shell 3.11.5
# 使用 `set -e` 让脚本在任何命令失败时立即退出，这是一个好习惯。
set -e
ENV_NAME="swv2"

echo "--- Creating virtual environment: $ENV_NAME ---"
# 使用系统默认的 python3 创建环境
uv venv $ENV_NAME -p python3

# 创建需求文件
echo "--- Generating requirements files ---"

# PyTorch (Nightly) - 只包含包名
cat > requirements.torch.txt << EOL
torch
torchvision
torchaudio
EOL

# Application dependencies - 只包含包名
cat > requirements.app.txt << EOL
albumentations
opencv-contrib-python
jupyter
matplotlib
lightning
wandb
einops
pandas
scikit-learn
datasets
transformers
kornia
bitsandbytes
peft
timm
openvino
nncf
lightgbm
triton
torchmetrics
ultralytics
heavyball
rich
seaborn
EOL
echo "--- Activating environment and installing packages ---"
# 使用 POSIX 兼容的点号激活
. "$ENV_NAME/bin/activate"

# 使用 uv 安装，将所有选项放在命令行
echo "--- Installing PyTorch (Nightly) ---"
uv pip install --pre \\
  --index-url <https://download.pytorch.org/whl/nightly/cu126> \\
  -r requirements.torch.txt

echo "--- Installing other application dependencies ---"
uv pip install \\
  --index-url <https://pypi.tuna.tsinghua.edu.cn/simple> \\
  -r requirements.app.txt

echo "--- Cleaning up requirements files ---"
rm requirements.torch.txt requirements.app.txt

echo "--- Setup complete. Environment '$ENV_NAME' is ready. ---"
echo "To activate it in your current shell, run: source $ENV_NAME/bin/activate"