2024/07 4

[Pytorch] CUDA 설치하기

my gpu : GeForce RTX 3090 우리가 pytorch 돌리기 위해서 설치해야 할 것은 3개이다.  NVIDIA 드라이버: 시스템 전체에 설치CUDA Toolkit: 시스템 전체에 설치cuDNN : 각 가상환경에 설치되며, 가상환경마다 독립적으로 관리됨중요한 사실 : 파이토치는 cuda 버전 11.8, 12.1 밖에 지원을 안한다. 그래서 사실 gpu가 뭔지보다 pytorch 돌릴거면 저 cuda 버전에 맞는 소프트웨어들을 설치하는게 중요하다. 못믿겠음 여기서 확인 ㄱㄱ https://pytorch.kr/get-started/locally/ 1. Windows 111. nvidia driver 설치 마법사https://www.nvidia.com/ko-kr/drivers/details/1843..

Python 2024.07.19

[Huggingface] clinicalBERT 모델을 이용한 의료상황분석

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline# 모델과 토크나이저 로드model_name = "emilyalsentzer/Bio_ClinicalBERT"tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2) # num_labels은 분류할 라벨의 수# 분류 파이프라인 설정classifier = pipeline('text-classification', model=model, tokenizer=tokenizer, ..

카테고리 없음 2024.07.17

[Huggingface] DistilBert 모델을 이용한 감성분석

from transformers import DistilBertTokenizer, DistilBertForSequenceClassificationfrom transformers import TextClassificationPipelineimport torch# 모델과 토크나이저 로드model_name = "distilbert-base-uncased"tokenizer = DistilBertTokenizer.from_pretrained(model_name)model = DistilBertForSequenceClassification.from_pretrained(model_name)# 파이프라인 설정classifier = TextClassificationPipeline(model=model, tokenizer..

카테고리 없음 2024.07.17