본문 바로가기

카테고리 없음

우분투 딥러닝 환경 구축, Enviornment costruction of deep learning for Ubuntu

1. Nvidia 드라이버 설치

ubuntu-drivers devices 이게 안되면 nvidia-smi 를 치면 가이드를 해줌.

sudo apt-get install nvidia-driver-510 server (필자는 이걸로 함)sudo apt-get remove --purge nvidia* (드라이버 삭제법)sudo apt-get autoremove (이것도가능)

sudo ubuntu-drivers autoinstall(자동) -정안되면 이걸로

reboot

2. CUDA설치 ( CUDA setup)

https://developer.nvidia.com/cuda-toolkit-archive

필자는 최신버전  CUDA Toolkit 11.6.0 (May 2022), https://developer.nvidia.com/cuda-11-6-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local 으로 받음

 

CUDA Toolkit 11.6 Downloads

Resources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverables

developer.nvidia.com

윈도우용 리눅스 용이 있으니 확인후 다운로드

파이토치를 사용하기 위해선 11.6으로 다운

Lunux-> x86_65 -> Ubuntu ->22.04 -> deb(local)

 

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb

sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

 

3. CUDNN 설치 (CUDNN setup)

NVIDIA 아이디 비번 필요

https://developer.nvidia.com/rdp/cudnn-download