for tensorflow 1.x 1080ti


[install graphic drivers]

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

$ sudo apt-get update

$ sudo apt-get install nvidia-390


$ sudo reboot


[check]

nvidia-smi


[install cuda]

sudo sh cuda_8.0.61_375.26_linux.run

### cuda 9.0으로 수정해서 깔아야함


Do you accept the previously read EULA?

accept/decline/quit: accept


Install NVIDIA Accelerated Graphics Driver for Linux?

(y)es/(n)o/(q)uit: n


Install the CUDA 9.0 Toolkit?  

(y)es/(n)o/(q)uit: y


Enter Toolkit Location  

 [ default is /usr/local/cuda-8.0 ]: 


Do you want to install a symbolic link at /usr/local/cuda?  

(y)es/(n)o/(q)uit: y


Install the CUDA 8.0 Samples?  

(y)es/(n)o/(q)uit: n


Enter CUDA Samples Location  

 [ default is /home/your_id ]: 




[set env]

$ echo -e "\n## CUDA and cuDNN paths"  >> ~/.bashrc

$ echo 'export PATH=/usr/local/cuda-9.0/bin:${PATH}' >> ~/.bashrc

$ echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:${LD_LIBRARY_PATH}' >> ~/.bashrc

$ source ~/.bashrc


[check]

$ nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2016 NVIDIA Corporation

Built on Tue_Jan_10_13:22:03_CST_2017

Cuda compilation tools, release 8.0, V8.0.61



[install cudnn]

### cuda 9.0에 맞게 깔아야


이후 pip를 install한 다음에

sudo pip install tensorflow

sudo pip install tensorflow-gpu

sudo pip3 install tensorflow

sudo pip3 install tensorflow-gpu


ubuntu 18.04사용하면서

python3에서 tensorflow를 불러오는게 안될 수 있음

이때 오류에 나온 pip.py인가를 띄워서 __internal을 추가하는거였나...(체크필요)





+ Recent posts