mirror of
https://github.com/wassname/discovering_latent_knowledge.git
synced 2026-09-24 13:20:12 +08:00
417 B
417 B
# note
conda create -n dlk2 python=3.9 -y
conda activate dlk2
mamba install -y pytorch torchvision torchaudio pytorch-cuda=11.7 cudatoolkit-dev==11.7 cudatoolkit=11.7 -c pytorch -c nvidia -c conda-forge
mamba install -y ipykernel pip
pip install -r requirements.txt
2023-05-13 15:17:05
- Convert it to lightning
- batch for get hidden states
- and cache
- 9s vs 60. so 10x faster