PyTorch 시작

이다언·2024년 12월 17일

PyTorch

목록 보기
1/3
post-thumbnail

환경설정

  1. 버전충돌방지위해 "pytorch_test" 라는 이름의 가상환경생성
    • conda create --name pytorch_test --clone base
  2. pytorch_test 가상환경으로 들어가기
    • conda activate pytorch_test
  3. pytorch cpu 버전 설치 (GPU 없을 때)
    • conda install pytorch=2.0.0 torchvision=0.15.0 cpuonly -c pytorch
  4. python 명령어 실행 후 import torch 실행하면 PyTorch 정상설치여부 확인가능
  5. 잘못만든 가상환경 지우기 conda remove --name [가상 환경 이름] --all
  • CUDA: GPU에서 사용하는 수천 개의 코어를 활용하여 병렬로 코드를 실행하게 해주는 기술
profile
AI 클라우드 웹개발자

0개의 댓글