Naver-AI-Hackathon / AI-Vision

67 stars 34 forks source link

nsml 세션 실행시 tensorflow와 pytorch 같이 사용가능한가요? #83

Closed yongjun823 closed 5 years ago

yongjun823 commented 5 years ago

세션: Vision_Magicians/ir_ph1_v2/41

/bin/bash: line 1: 49 Segmentation fault (core dumped) python -u nsml_main.py

tensorflow와 pytorch 모두 import 하면 오류가 발생하는데 하나만 사용할 수 있나요?

#nsml: nsml/ml:cuda9.0-cudnn7-tf-1.11torch0.4keras2.2

setup.py에 기본 이미지 사용하고 있습니다. 감사합니다.

yongjun823 commented 5 years ago

nsml: nsml/ml:cuda9.0-cudnn7-tf-1.11torch1.0

도커 이미지 변경하니 잘됩니다