theo.log
로그인
theo.log
로그인
TensorFlow, PyTorch에서 GPU 연동
Theo Kim
·
2023년 3월 1일
팔로우
0
PyTorch
tensorflow
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를 활용하도록 설정
Theo Kim
THEO's velog
팔로우
이전 포스트
ML Project1(모델, 결과, 결론, 후기)
다음 포스트
DL Project (파손 상품 분류)
0개의 댓글
댓글 작성