因tensorflow版本的原因,现需要卸载原来的cuda8.0版本,重新安装cuda9.0版本。
一、卸载cuda8.0
使用cuda自带的卸载工具进行卸载。
cd /usr/local/cuda/bin
sudo ./uninstall_cuda_9.0.pl
安装指示卸载即可。
二、安装cuda9.0
2.1 安装驱动
首先安装驱动。
最简单的办法是点击Applications-->System-->Software Updater。
选择Additional Drivers选项卡。
选择要应用的驱动版本,点击Apply Changes,等待驱动安装好,安装好之后需要重启电脑。
2.2 安装cuda9.0
执行
sudo sh cuda_9.0.176_384.81_linux.run
一路按照指示安装即可。只是有一个地方需要注意:在询问是否安装gpu drivers时要选择no。
安装路径在/usr/local/cuda-9.0,一般会链接到/usr/local/cuda。安装成功之后,还要配置环境变量。
执行
vim ~/.bashrc
在末尾添加如下内容:
export PATH="/usr/local/cuda/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
执行
sudo vim /etc/profile
在末尾添加如下内容。
export PATH=/usr/local/cuda/bin:$PATH
export CUDA_INSTALL_PATH=/usr/local/cuda
export LD_LIBRARY_PATH=$CUDA_INSTALL_PATH/lib64:$LD_LIBRARY_PATH
执行
source /etc/profile
使设置生效。
创建链接文件:
sudo vim /etc/ld.so.conf.d/cuda.conf
在其中添加如下语句:
/usr/local/cuda/lib64
然后执行
sudo ldconfig
2.3 安装cudnn
下载cudnn需要登录nvidia账户,请自行下载。
下载之后解压,目录结构如下:
将头文件和库文件分别拷贝到cuda对应的目录下:
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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
最后更新一下链接:
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
查看一下nvcc的信息验证安装是否成功。
nvcc -V