IT개발도상국

라이브러리 사용 본문

setup

라이브러리 사용

블랙헤이돈 2023. 6. 5. 18:37

pytorch 

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

 

파이토치 한국 사용자 모임 (PyTorch Korea User Group)

파이토치 한국 사용자 모임에 오신 것을 환영합니다. 딥러닝 프레임워크인 파이토치(PyTorch)를 사용하는 한국어 사용자들을 위해 문서를 번역하고 정보를 공유하고 있습니다.

pytorch.kr

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

아나코다에 설치할 가상환경에 붙여넣기 

 

 

CUDA 설치 

https://developer.nvidia.com/cuda-toolkit-archive

 

CUDA Toolkit Archive

Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production

developer.nvidia.com

 

버전에 맞는 pytorch 선택하여 다운로드 11.8 버전의 cuda를 설치하여서 11.8인거 설치 

 

cuDnn 설치 

https://developer.nvidia.com/rdp/cudnn-archive

 

cuDNN Archive

Download releases from the GPU-accelerated primitive library for deep neural networks.

developer.nvidia.com

CUDA 버전과 맞는것 설치후 내용을 복사하여 CUDA파일에 넣으면 끝남

 

Comments