Miniconda 개발 환경 설치

다음 포스트를 참고하여 Miniconda 개발환경을 설치하세요. 파이썬 프로젝트 별로 패키지를 따로 관리할 수 있습니다.

Visual Studio Code와 Miniconda를 사용한 Python 개발 환경 만들기( Windows, Ubuntu, WSL2)

PyTorch 설치

아래 링크에 접속하여 Your OS는 Linux, Package는 Conda, Language는 Python, Compute Platform으로 원하는 CUDA 버전을 선택하면 스크린샷처럼 Pytorch를 설치에 사용할 수 있는 명령어가 보입니다.

현재 2025년 4월 27일 기준으로 보이는 선택항목이며 나중에 바뀔 수 있습니다.

1. conda를 사용하여 PyTorch를 위한 Python 가상 환경을 생성합니다.

$ conda create -n pytorch python=3.12

y 를 입력 후 엔터를 누릅니다.

Proceed ([y]/n)?

2. python 가상 환경을 활성화합니다.

$ conda activate pytorch

3. 위에서 CUDA 버전을 선택하여 확인한 다음 명령을 사용하여 GPU를 사용하는 PyTorch 패키지를 설치합니다. CUDA 12가 같이 설치되기 때문에 따로 CUDA를 설치할 필요가 없습니다. 글작성 시점에서 PyTorch 2.7.0이 설치됩니다.

$ pip3 install torch torchvision torchaudio

4. 설치 완료후 GPU 사용가능한지 체크합니다. 마지막에 True가 나와야 합니다.

$ python3 -c "import torch; print(torch.cuda.is_available())"

True