https://developer.nvidia.com/cudnn
NVIDIA cuDNN
NVIDIA cuDNN The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, nor
developer.nvidia.com
로컬에서 GPU를 사용하기 위해 cuda를 다운받는다.
장치관리자에 어떤 GPU를 쓰고 있는지 확인하고 GPU를 학습에 쓸 수 있는지 알아낸다.
cnDNN에서 자신에게 맞는 zip 파일을 다운 받는다.
이 zip 파일을 보면 세가지 폴더가 있다.
이 세가지 폴더를 내가 만든 가상환경에, 또는 로컬 conda안에 설치한다.
Lib -> site-packages -> numba -> cuda
이 cuda 폴더에 압축을 푼다.
나는 가상환경을 사용해서 일단 실행을 시켜주고
https://pytorch.org/get-started/locally/
여기 사이트에서 pytorch를 위한 설치 run this command를 실행시켰다.
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
'포트폴리오 Portfolio > AI Project' 카테고리의 다른 글
| 혐오 발언 식별 다썰어 프로젝트 (0) | 2022.05.25 |
|---|---|
| 방법론 조사 (0) | 2022.04.13 |
| 모델링 방법 조사 BinaryClassification, Multi label, Multi class Classification (0) | 2022.04.12 |
| AI Project NLP 악플 데이터 셋 일부 proprocessing eda (0) | 2022.04.01 |
| 악플 데이터 셋 추가 (0) | 2022.03.30 |