安装

在开始之前，您需要配置环境并安装相应的软件包。timm 已在 Python 3+ 上经过测试。

虚拟环境

建议在虚拟环境中安装timm，以保持环境整洁并避免依赖冲突。

1. 创建并进入您的项目目录：

   已复制

   mkdir ~/my-project
   cd ~/my-project

2. 在您的目录中启动一个虚拟环境：

   已复制

   python -m venv .env

3. 使用以下命令激活和取消激活虚拟环境：

   已复制

   # Activate the virtual environment
   source .env/bin/activate
   
   # Deactivate the virtual environment
   source .env/bin/deactivate

创建虚拟环境后，您可以在其中安装 timm。

使用 pip

安装 timm 最直接的方式是使用 pip：

pip install timm

或者，您可以直接从 GitHub 安装 timm 以获取最新的开发版本：

pip install git+https://github.com/rwightman/pytorch-image-models.git

运行以下命令以检查 timm 是否已正确安装：

python -c "from timm import list_models; print(list_models(pretrained=True)[:5])"

此命令会列出 timm 中可用的前五个预训练模型（按字母顺序排序）。您应该会看到以下输出：

['adv_inception_v3', 'bat_resnext26ts', 'beit_base_patch16_224', 'beit_base_patch16_224_in22k', 'beit_base_patch16_384']

从源代码

从源代码构建 timm 允许你对代码库进行修改。要从源代码安装，请克隆仓库并使用以下命令进行安装：

git clone https://github.com/rwightman/pytorch-image-models.git
cd pytorch-image-models
pip install -e .

同样，你可以使用以下命令检查 timm 是否已正确安装：

python -c "from timm import list_models; print(list_models(pretrained=True)[:5])"