File size: 535 Bytes
303c2e0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia
pip install packaging
pip install ninja
pip install transformers
pip install datasets
pip install omegaconf
conda install ipykernel
python -m ipykernel install --user --name tr2d2 --display-name "Python (tr2d2)"
pip install hydra-core --upgrade
pip install hydra-submitit-launcher
# for mdlm
pip install causal-conv1d
pip install lightning
pip install timm
pip install rich
pip install scipy
pip install wandb
pip install gReLU |