WSL 사용해서 윈도우(VSCODE)에 Transformer 학습 셋팅 하기

 

conda update -n base -c defaults conda

conda create -n python311 python=3.11

conda activate python311

pip3 install torch torchvision torchaudio

 

VSCODE에서 

ctrl + shift + p 

wsl:connect to wsl

이렇게 하면 연결 됨

 

vscode 안에서 jupyter notebook 환경도 실행해 볼 수 있음

 

requirements.txt 에 이런 것들 추가해서 설치 하고 실행

torch

transformers==4.41.2

datasets==2.18.0

quanto

peft==0.11.0

trl==0.8.6

accelerate==0.31.0

sentencepiece

ipywidgets

sentence-transformers