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개의 댓글
댓글 작성
관련 채용 정보