티스토리 뷰

반응형

다음 사이트에 접속해서 컴퓨터 환경에 맞는 설치 명령어를 생성한다.

 

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

 

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

해당 컴퓨터 환경은 cuda 11.8을 설치한 경우이다. 최종적으로 생성구문은 다음과 같다.

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

가상환경 tf39로 접속해서 위의 설치구문을 실행한다.

pytorch를 성공적으로 설치 후 

쥬피터 노트북을 실행하여 GPU를 인식하는지를 확인한다.

GPU를 인식하는 실행 코드는 다음 2가지가 있다.

 

 

반응형
댓글