Deep learning/환경세팅

Ubuntu 환경 세팅

비비이잉 2021. 8. 6. 17:15
반응형

ubuntu 20.04 Nvidia 그래픽 카드 드라이버 설치

 

sudo ubuntu-drivers devices


sudo ubuntu-drivers autoinsatll

sudo apt isntall nvidia-drivers-460 
#사용하는 그래픽 카드에 따라 drivers-버전을 설치해야함 

nvidia-smi
#이 명령어 쳤을 때 그래픽 드라이버가 떠야 정상 
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is insatlled and running. 

오류가 뜬다. 다시 깔았던 버전 지우고 깔고 reboot 하면 해결

 

apt --installed list | grep nvidia-driver

sudo apt remove nvidia-driver-버전
sudo apt automove

sudo apt-get install nvidia-driver-버전
sudo reboot now 
#

 

 

 

python 패키지 설치

#중간중간에 sudo apt-get update, sudo apt-get upgrade 는 계속 해줘야함 

pip3 install numpy
pip3 install matplotlib pandas tensorflow cv2 pytorch
#tensorflow-2.5.0 
#pandas-1.3.0 
#matplotlib-3.4.2 
#cv2


 

 

CUDA toolkit ubuntu

sudo apt update
sudo apt insatll nvidia-cuda-toolkit

 

 

cuda 버전에 따라서 호환되는 cuDNN 버전이 다르다.

#ubuntu architecture 확인 
dpkg -s libc6 | grep Arch
#Architecture : amd64
#Multi-Arch : same

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=deb_local

 

 

반응형

'Deep learning > 환경세팅' 카테고리의 다른 글

server setting 서버 세팅 ulimit  (0) 2022.11.22
[Error] failed with initial frozen solve  (0) 2021.10.18
Gitlab ssh-key  (0) 2021.08.06