Skip to content

Latest commit

 

History

History
76 lines (50 loc) · 1.92 KB

timm09_04.md

File metadata and controls

76 lines (50 loc) · 1.92 KB

安装

原始文本:huggingface.co/docs/timm/installation

在开始之前,您需要设置您的环境并安装适当的软件包。timm在**Python 3+**上进行了测试。

虚拟环境

您应该在virtual environment中安装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 timm
pip install -e .

同样,您可以使用以下命令检查是否已正确安装timm

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