Unbuntu 16.04 Tensorflow重新安装

1. 卸载原CUDA版本

https://blog.youkuaiyun.com/u012074597/article/details/80317275?utm_source=blogxgwz2

这里以CUDA10卸载为例子

sudo apt-get autoremove --purge cuda
sudo dpkg -l |grep cuda
sudo dpkg -P cuda-repo-ubuntu1604-10-1-local_10.1.85-1_amd64
sudo apt-key list
sudo apt-key del 7fa2af80

2. 下载新CUDA 8.0

https://developer.nvidia.com/cuda-80-ga2-download-archive

运行三条安装命令即可

安装完成后

sudo gedit ~/.bashrc

在文件末尾添加
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

再运行

source ~/.bashrc

 

3. 下载安装CUDNN

https://developer.nvidia.com/rdp/cudnn-archive

我选择的是for Linux

下载完成后解压

tar xvzf cudnn-8.0-linux-x64-v6.0-ga.tgz

然后

sudo cp cuda/include/cudnn.h  /usr/local/cuda/include 

sudo cp cuda/lib64/libcudnn*  /usr/local/cuda/lib64 

sudo chmod a+r /usr/local/cuda/include/cudnn.h/  /usr/local/cuda/lib64/libcudnn*  

 

4. 重新安装Tensorflow

pip uninstall tensorflow-gpu

pip install tensorflow-gpu==1.3.0

安装完成