Naver-AI-Hackathon / AI-Speech

31 stars 5 forks source link

NSML 환경에 대해 질문드립니다. #24

Open qute012 opened 4 years ago

qute012 commented 4 years ago

혹시 pytorch 버전이 몇 버전인지 알 수 있을까요?

nsml-admin commented 4 years ago

설정하신 환경에 따라서 달리질수 있을것 같습니다. 기본 "nsml/default_ml:latest" 환경을 쓰신다면 https://ai.nsml.navercorp.com/support 여기서 확인할 수 있듯이 pytorch 0.3 version을 사용하며, support 페이지에 제공된거 이외 환경을 사용하고 싶으시면 아래 페이지에서 나온 방법을 통해 다른 환경을 사용 할 수 있습니다.

https://n-clair.github.io/ai-docs/_build/html/ko_KR/contents/session/prepare_a_session_environment.html

ClovaAIAdmin commented 4 years ago

추가로 말씀드리면, 세션 로그를 보시면 상단에 pytorch version이 출력되어서 확인 가능하십니다.

HanSeokhyeon commented 4 years ago

`#nsml: nvcr.io/nvidia/pytorch:19.06-py3 from distutils.core import setup import setuptools

setup( name='speech_hackathon', version='1.0', install_requires=[ 'python-Levenshtein' ] )`

보시는 것과 같이 기본 코드는 nvcr.io/nvidia/pytorch:19.06-py3를 쓰고 있네요. pytorch:19.06은 pytorch 1.12.0 사용합니다.