装好ubuntu后,不需要安装显卡驱动,因为cuda有配套的显卡驱动。切记:只限于单显卡的用户
0、安装显卡驱动
上图来自:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
添加 ppa:
sudo add-apt-repository ppa:graphics-drivers/ppa (注意联网,去掉代理)
sudo apt update
1、安装cuda
sudo dpkg -i cuda-repo-ubuntu1604-10-0-local_10.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-10.0
2、安装cudnn
tar xzvf cudnn-10.0-linux-x64-v7.1.tgz
cd cuda
sudo cp include/* /usr/local/cuda-10.0/include
sudo cp lib64/* /usr/local/cuda-10.0/lib64
在终端执行gedit .bashrc ,在文件.bashrc最后一行加上以下:
export PATH="/usr/local/cuda-10.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LIBRARY_PATH"
export CUDA_HOME=/usr/local/cuda
关机重启
3、安装anaconda
sh Anaconda3-5.0.0.1-Linux-x86_64.sh
一路回车键就行了。
4、安装tensorflow和keras
pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install msgpack -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install keras==2.2.6 -i https://pypi.tuna.tsinghua.edu.cn/simple/