반응형 텐서플로우2 [ 자취방 워크스테이션 ] #20 도커 컨테이너의 GPU사용 확인하기 //1 이번 글은 아주 간단하지만 대단히 중요한 부분이라서 따로 글을 쓴다. 바로 GPU사용에 관한 것인데 파이토치에서 사용하는 방법도 있으나 일단은 tensorflow를 기준으로 하겠다. //2 이전 글까지 따라왔다면 주피터를 잘 사용하고 있을 것이다. 이제는 주피터에 들어가서 터미널을 하나 열고 아래와 같이 입력해주자. $ nvidia-smi 이렇게 생긴 표가 나오면 성공이다. 이 표는 ====선을 기준으로 선 위에 적힌 내용에 해당하는 값이 선 아래의 같은 위치에 위치해 있다. 이번엔 옵션을 하나 줘서 $ nvidia-smi -l 1 이렇게 하면 계속 이 표가 갱신된다. -l 은 루프를 의미하고 1은 1초마다 한번씩이다. 이 상태에서 Ctrl+C로 멈출 수 있다. 여기서 주피터노트북을 열고 아래와 .. 2021. 3. 20. [ Python 삽질 ] __call__() got an unexpected keyword argument 해결법 //서론 파이썬의 클래스는 콜 메소드(맴버 함수)를 가지고 있을 수 있다. 만약 누군가 만들어둔 라이브러리(keras라던가...)를 사용할 때 이 콜 메소드를 함부로 수정하게 되면 수정된 클래스를 단독으로 사용할 때는 문제가 크게 안되지만 다른 모듈과 함께 사용할 때는 문제가 될 수 있다. 나도 어제 오래된 논문의 코드를 하나 실행시키다가 저자가 loss function을 수정해 둔 것을 발견하지 못하고 삽질을 해 버렸다. 자유도가 낮은 코드일 수록 버전이 바뀌면 이런 자잘한 문제들이 발생하기 쉽다. 이 문제는 굉장히 간단한 문제이지만 keras 코드상에서 막상 직면하면 model.fit() 과 model.compile() 안에서 동작하므로 번뜩 떠오르지 않는 디버깅일 수 있다. 따라서 정리해 놓고자 한.. 2020. 12. 29. 이전 1 다음 반응형
