인공지능

[Windows] Python + Pytorch 개발 환경 세팅하기 (feat. Vscode)

Shong Studio 2024. 6. 3. 00:38
728x90
반응형

1. 명령 프롬프트 창 -> winget install -e --id Python.Python.3.10

 

2. 잘 설치되었는지 확인 (새로운 명령 프롬프트 창을 다시 열어야합니다.)

 

3. PyTorch 2.1.2 설치 : pip3 install torch torchvision torchaudio --index-url  https://download.pytorch.org/whl/cu121
    PyTorch 삭제 : pip3 uninstall torch torchvision torchaudio

 

4. 아래와 같이 pytorch 버전과 cuda 사용 여부를 확인할 수 있습니다.

 

5.  VScode에서 파일 만들어서 테스트

 

기본적인 Python 작업 개발환경 세팅을 마쳤습니다.

728x90
반응형