AI를 이용한 노트 필기 앱 : APlus (GPU 서버 이용)

danbibibi·2022년 1월 18일
0

✔️ GPU 환경 설정

접속방법
리눅스 기본 명령어
주피터 노트북 원격 서버 설정
주피터 노트북 원격 서버 설정

✔️ GPU 서버 접속

학교에서 캡스톤 시 원하면 gpu 서버를 할당해준다. gpu서버 말만 들어봤지 실제로 사용해본 적이 한 번도 없어서 아주 모르는 것 투성이다. 우선 gpu서버에 접속하기 위해서는 아래와 같이 접속하면 된다고 한다! 교외에 있을 때는 vpn 신청 후 접속해야한다. 처음에 timeout 떠서 여쭤보니 방화벽(?)이 있어서 그렇다고 한다.

ssh [사용자 계정]@[원격지 ip]

아무튼 vpn 신청 후 gpu 서버에 접속해보니 python, anaconda 등 기본적인 것은 깔려 있었다. jupyter notebook을 이용하기 위해서는 이 게시물을 참고 했다.

✔️ GPU 정보 확인

아래 명령어를 통해서 gpu가 제대로 올라왔는지와 관련 상세 정보를 확인할 수 있다.

$nvidia-smi

✔️ PyTorch gpu 설정

# 사용이 가능한 경우 True 반환
print(torch.cuda.is_available())

PyTorch에서 다음과 같이 gpu를 설정할 수 있다.

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

위와 같이 gpu 설정 후에는 .to(device)를 사용하여 model, data 등을 GPU에 넣어 사용하면된다.

# example
model.to(device)
features.to(device)
labels.to(device)
profile
블로그 이전) https://danbibibi.tistory.com

0개의 댓글