Skip to content

Setup Environment

We use poetry as the virtualenv as well as project manager (e.g. dependencies, packaging, publishing, etc.).

Please read about poetry (see REFERNCES section at the end of this page) and the pyproject.toml file before you run following commands to setup your local develop environment. Having another pip installed release version of ice will not cause a problem since poetry isolates the environment.

Pypi Mirror for China (Optional)

Aliyun

mkdir -p ~/.pip
echo "[global]\nindex-url = https://mirrors.aliyun.com/pypi/simple/\n[install]\ntrusted-host=mirrors.aliyun.com\n" > ~/.pip/pip.conf

Douban

mkdir -p ~/.pip
echo "[config]\nindex-url = https://pypi.douban.com/simple" > ~/.pip/pip.conf

Steps

  1. Install poetry following the instruction here.
  2. Set tab-completion for poetry following the instruction here
  3. git clone https://github.com/tjyuyao/ice-learn
  4. cd ice-learn
  5. poetry shell
  6. poetry install -E pycuda
  7. Set tab-completion for poe-the-poet following the instruction here.
  8. Install torch and torchvision manually for correct cuda version using pip in the poetry shell, e.g.: pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

References