mirror of
https://github.com/wassname/discovering_latent_knowledge.git
synced 2026-09-24 13:20:12 +08:00
16 lines
417 B
Markdown
16 lines
417 B
Markdown
```sh
|
|
# 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
|
|
|
|
- [x] Convert it to lightning
|
|
- [ ] batch for get hidden states
|
|
- [x] and cache
|
|
- [ ] 9s vs 60. so 10x faster
|