[중요] 우분투에 딥러닝 환경 구축 총정리

About_work·2023년 10월 2일
1

딥러닝 환경 구축

목록 보기
3/23

왜 이 글을 읽어야 하는가?

  • 우분투에 딥러닝 환경 구축을, 어떤 컴퓨터에서도 할 수 있게 하기 위함.
  • 모든 과정을 이해하고, 확신을 가지고 설치하기 위함.

내 생각 정리

  • 내 nvidia gpu 에 맞는 버전의 nvidia driver을 설치
  • 설치한 nvidia driver 버전에 맞는 버전의 cuda toolkit 설치
  • 설치한 cuda 버전에 맞는 pytorch 버전 설치

사전 개념


방법

내 pc GPU 확인


내 nvidia gpu 에 맞는 버전의 nvidia driver을 설치


virtualenv 설치


내 GPU & 설치한 nvidia driver 버전 & torch 버전에 맞는 버전의 cuda toolkit 설치


pytorch 설치


설치 후, cuda toolkit 버전 확인 방법


cuDNN 설치


profile
새로운 것이 들어오면 이미 있는 것과 충돌을 시도하라.

0개의 댓글