카테고리 없음

ubuntu 18.04 cuda 8 install for tensorflow 1.4.1

hellobird 2019. 8. 20. 22:44

https://askubuntu.com/questions/1028830/how-do-i-install-cuda-on-ubuntu-18-04



94

In a terminal, type:


sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt update

sudo ubuntu-drivers autoinstall

reboot


sudo apt install nvidia-cuda-toolkit gcc-6

nvcc --version
checking : nvcc --version




cudnn for cuda-8 : 
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.1.3/prod/8.0_20180414/cudnn-8.0-linux-x64-v7.1

--------------------------------------------
tar xzvf cudnn-8.0-linux-x64-v7.1.tgz
sudo cp cuda/lib64/* /usr/lib/cuda/lib64/
sudo cp cuda/include/* /usr/lib/cuda/include/
sudo chmod a+r /usr/lib/cuda/lib64/libcudnn*
sudo chmod a+r /usr/lib/cuda/include/cudnn.h

Checking : cat /usr/lib/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2