tensorflow에 필요한 각종 라이브러리 버전 정보

ubuntu 18.04 LTS

 

텐서플로 설치안내문 :https://www.tensorflow.org/install/gpu

텐서플로와 gpu 

 

1. nvidia_driver는 418버전으로
sudo apt-get install nvidia-driver-418

2. cuda 10.1 설치 

https://developer.nvidia.com/cuda-toolkit-archive
sudo ./cudnn10.1~~.run
nvcc가 /usr/loca/cuda-10.1 에 설치되는데, 이 nvcc을 /usr/bin 같은데로 link 걸지말자. 
cudnn에서 nvcc 패스를 찾아서 설치하는데 잘못된 곳에 설치할수 있다

 

export PATH=$PATH:/usr/local/cuda/bin 

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

을 .bashrc같은 곳에

3. cudnn 7.6.5 runtime부터 설치하고 dev설치

https://developer.nvidia.com/rdp/cudnn-download
sudo dpkg -i cudnn-7.~~
sudo dpkg -i cudnn-dev

 

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64

을 .bashrc같은 곳에

 

4. tensorflow 는 2.1버전 설치
virtualenv 안에 gpu버전으로 설치

 

tensorflow 실행해 보면 tensorrt 관련 에러들이 뜰수있음. 

https://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html

TensorRT Installation Guide :: NVIDIA Deep Learning SDK Documentation

https://developer.nvidia.com/nvidia-tensorrt-7x-download

불러오는 중입니다...

tensorRT는 cuda 10.0 , 10.2만 지원 

10.1을 쓰려면 사용 못함