TensorFlow, PyTorch에서 GPU 연동

Theo Kim·2023년 3월 1일
0

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개의 댓글