파이토치 GPU 버전 설치하기

1. 다음 URL로 이동하면 다양한 환경에 맞는 명령어를 찾을 수 있습니다.

https://pytorch.org/get-started/locally/

이때 중요한 것은 다음 그림과 같이 반드시 ‘Compute Platform’ 부분에서 CUDA를 선택해야 한다는 것입니다.

▲ 그림 8-30 버전 선택

2. 다음 명령을 아나콘다 프롬프트에서 실행합니다.

먼저 예제를 진행하고 있는 가상 환경으로 이동한 후 다음 명령을 실행합니다. 명령은 선택한 환경에 따라 다를 수 있습니다.

> conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

이제 파이토치에서 GPU를 사용할 수 있는 환경이 구성되었습니다. 파이토치는 GPU에 최적화되었기 때문에 GPU 환경이 가능하다면 GPU 위주로 사용하면 좋습니다.