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