놀고 싶은 빈둥이의 coding 연구소

torch.nn.modules.module.ModuleAttributeError: 'Model' object has no attribute '_non_persistent_buffers_set' python에서 pytorch 라이브러리를 이용해 딥러닝 모델을 만들고 해당 모델을 돌릴 때 발생하는 에러다. 알아보니 pytorch 버전이 1.6.x일 때 발생하는 에러다. pytorch 버전을 1.5.1로 downgrade하면 해당 에러가 해결된다. python 버전이 3.8로 상향되면서 pytorch에서 라이브러리를 설치할 때 자동으로 1.6.x 버전이 설치된다. 때문에 pytorch 설치 후 다시 downgrade를 해야 해당 에러가 발생하지 않는다. pytorch를 downgrade하는 방법은 pytorc..

개발 환경 Pycharm Python: 3.7 pytorch: 1.3.1 cuda: 10.2 문제: MLP 설계 시에는 문제 없이 돌아가는데 CNN으로 하는 경우 위 해당 코드가 나오면서 코드 실행이 중단된다. 시도(해당 에러 코드를 검색해보면 메모리 접근 위반이라는데 이렇다할 해결 방법은 찾지 못했다.) - CUDA-CuDNN를 다시 설치해도 해결되지 않는다. - Tensorflow를 이용한 CNN 코드를 Pycharm에서도 돌리면 문제없이 작동한다. - 같은 코드를 colab에서 돌리면 정상적으로 작동한다. - 같은 코드를 jupyter notebook에서 돌리면 정상적으로 작동한다. 결론(미해결) -> 코드에는 문제가 없어 보이고 Pycharm에서 설정 문제일 수도 있겠지만, 구글에 검색해도 나오지..