리눅스를 통한 환경 세팅

기존 ubuntu 16.04

cuda 10.01 cudnn 7.6.5 Nvidia driver 410.39

-> cuda 10.0 cudnn 7.4 Nvidia driver 410 (TF_1.14-gpu)

/home 메모리 부족으로

/home/han/study/han /home/han/dataset /home/han/virtualenv/torch041 /torch03 /torch031 을 /mnt/hdd640/dong_backup으로 이동

그래픽카드 도는지 확인

window에서 글카 도는지 확인 cd C:\Program Files\NVIDIA Corporation\NVSMI에서 nvidia-smi

우분투 1초마다 nvidia 모니터링 watch -n 1 nvidia-smi

환경세팅 요약

Nvidia driver, cuda, cudnn 각각 다운 받고 cudnn cuda의 include에 각각 옮기고,

vi로 bashrc에서 path 수정하기

bashrc 수정

  1. vi ~/.bashrc 켜고 i 눌러서 insert 모드

  2. 맨 밑줄에

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH 추가

  1. esc 누르고 : 누른 후 wq로 저장

  2. source ~/.bashrc 로 bashrc 업데이트

GPU 두 개로 tensorflow 학습하기

마찬가지로 bashrc에서 맨 밑줄에

export CUDA_VISIBLE_DEVICES=0,1 추가