TensorFlow, PyTorch에서 GPU 연동

Theo Kim·2023년 3월 1일

1. NVIDIA DRIVER 설치

2. CUDA Toolkit 설치

3. 환경 변수 추가 확인

4. cuDNN 설치

5. CUDA, cuDNN(버전은 무조건 동일하게!!)

conda install cudatoolkit=12.0  # 자신의 기기에 맞는 버전!
conda install -c conda-forge cudnn

6. 프로그램이 GPU를 활용하도록 설정

profile
THEO's velog

0개의 댓글