[파이썬 에러] 대상 컴퓨터에서 연결을 거부했으므로 연결하지 못했습니다

문제

Python 3.11.5
CUDA 11.8
pytorch 2.3.0

평소와 같이 학습 환경 세팅 끝내고 train.py를 딱 실행했는데...,.,? 

딥러닝 모델 공부하는 동안에는 한번도 보지 못했었던...! 오류가 떴다.

검색을 하니 다들 소켓 통신 관련 오류 해결법만 알려줘서 꽤나 골머리를 썩이고 있었다... 

 

원인

오류 코드를 따라가보니 문제는 visdom 라이브러리를 사용하는 코드 부분에서 발생한 것이었다.

내가 사용한 코드는 visdom는 파이토치로 구현된 모델의 학습 과정이 시각화되어서 로컬 웹서버에 보여지도록 만들어져 있었다.

처음부터 security 에러가 나서 서버를 끈 채로 학습을 돌렸는데, 그러다보니 학습 결과가 띄워질 display가 없어서 오류가 난 것이었다.

 

해결 방법

1. visdom를 사용하는 대신 tensorboard 라이브러리로 대체해서 사용하자.

2. 나의 경우엔 visdom 라이브러리를 사용하지 않고 log에 학습 과정이 기록되게 수정했다.

 

왜? -> 굳이 시각화해서 학습 결과를 관찰할 필요가 없던 모델이었기 때문이다.

아 혹시 도움이 될까바 덧붙이자면, 나의 경우에는 코드에 optparser로 display 번호를 적는 argument를 설정할 수 있었는데, 그냥 이 부분을 --display_id 0 으로 설정해서 실행시켰다.

 

---

 

혹시 궁금한 점이 있으시다면 댓글 남겨주세요 ❣️