PyTorch

PyTorch project 는 심층 학습 네트워크를 빌드하기 위한 GPU 가속 텐서 계산과 높은 수준의 기능을 제공 하는 Python 패키지입니다. 라이선스 세부 정보는 GitHub에서 PyTorch 라이선스 문서를 참조 하세요.

PyTorch 모델을 모니터링 하 고 디버그 하려면 TensorBoard를 사용 하는 것이 좋습니다.

PyTorch는 Machine Learning에 대 한 Databricks Runtime에 포함 되어 있습니다. Databricks Runtime를 사용 하는 경우 PyTorch 설치에 대 한 지침은 PyTorch 설치 를 참조 하세요.

참고

이는 PyTorch에 대 한 포괄적인 가이드가 아닙니다. 자세한 내용은 PyTorch 웹 사이트를 참조 하세요.

단일 노드 및 분산 학습

단일 컴퓨터 워크플로를 테스트 하 고 마이그레이션하려면 단일 노드 클러스터를 사용 합니다.

심층 학습을 위한 분산 학습 옵션은 분산 학습을 참조 하세요.

예제 Notebook

PyTorch 노트북

Notebook 가져오기

PyTorch 설치

ML에 대 한 Databricks Runtime

Machine Learning에 대 한 Databricks Runtime에 는 클러스터를 만들고 PyTorch를 사용 하 여 시작할 수 있는 PyTorch 포함 되어 있습니다. 사용 중인 Databricks Runtime ML 버전에 설치 된 PyTorch 버전은 릴리스 정보를 참조 하세요.

Databricks Runtime

Databricks Machine Learning에 대 한 Databricks Runtime에 포함 된 PyTorch을 사용 하는 것이 좋습니다. 그러나 Databricks Runtime를 사용 해야 하는 경우 Databricks pypi 라이브러리로PyTorch를 설치할 수 있습니다. 다음 예제에서는 PyTorch 1.5.0를 설치 하는 방법을 보여 줍니다.

  • GPU 클러스터에서 다음을 pytorch 지정 하 여 및를 설치 torchvision 합니다.

    • torch==1.5.0
    • torchvision==0.6.0
  • CPU 클러스터에서 pytorchtorchvision 다음 휠 파일을 사용 하 여 및를 설치 합니다.

    https://download.pytorch.org/whl/cpu/torch-1.5.0%2Bcpu-cp37-cp37m-linux_x86_64.whl
    
    https://download.pytorch.org/whl/cpu/torchvision-0.6.0%2Bcpu-cp37-cp37m-linux_x86_64.whl