Skip to content
This repository has been archived by the owner on Jan 8, 2025. It is now read-only.

Latest commit

 

History

History
36 lines (24 loc) · 711 Bytes

INSTALL.md

File metadata and controls

36 lines (24 loc) · 711 Bytes

Installation

The code was tested with Anaconda Python 3.8, CUDA 11.3, and PyTorch 1.10.1. After installing Anaconda:

  1. [Optional but highly recommended] create a new conda environment.

    conda create -n hier python=3.8

    And activate the environment.

    conda activate hier

  2. Install PyTorch:

    conda install pytorch=1.10.1 torchvision=0.11.2 cudatoolkit=11.3 -c pytorch

  3. Clone this repo:

    git clone https://github.com/anthonyweidai/HierAttn.git

  4. Install the requirements

    cd $HierAttn_ROOT
pip install -r requirements.txt