方法1:NVIDIA 显卡驱动安装示例(ubuntu系统刚装好-推荐)
ps:
1.新系统的签名要重新注册
2.要禁用nouveau(新安装方法)
其他参考:
https://zhuanlan.zhihu.com/p/63584709
方法2:cuda 10.1 安装示例
- tensorflow 与 cuda版本对照表
https://tensorflow.google.cn/install/source#linux - 或可视化查看
- https://developer.nvidia.com/cuda-downloads
准备材料
- 下载地址&安装教程:https://developer.nvidia.com/cuda-toolkit-archive
安装过程
cuda 10.1 安装示例:
#下载安装包
wget https://developer.download.nvidia.cn/compute/cuda/10.1/secure/Prod/local_installers/cuda_10.1.168_418.67_linux.run?2dTVCGN0mdtX4qJMix6nqZCwHYgyX347T7z_b72fY3W49UlWkXMSqyDlDCxKaZwaHsnw1dDnQeLMqgNUuktT9hrI35nC2O93ad9DW2E351tIBKHsfYQ91JuOcr83qsvD_dIbaaaX8K-XiHW4WckuQsQ_ksj_E5IntYZ_oJTk4VOcT-0MoEpZS7Suv5E
#安装
sudo sh cuda_10.1.168_418.67_linux.run
#参考:https://docs.nvidia.com/cuda/cuda-quick-start-guide/index.html#ubuntu-x86_64-run
# 配置bash
vim ~/.bashrc
# 将下面的内容填进~/.bashrc中
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=/usr/local/cuda-10.1
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64:/usr/local/cuda-10.1/lib
# 激活
source ~/.bashrc
conda安装cudnn
conda install tensorflow-gpu=1.14.0 cudatoolkit=10.0 cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/