有个单卡的服务器,想在上面装tensorflow 1.0,但是报错提示需要cuda 8.0,而本机的cuda是7.0.
于是就想进行升级:
升级之前的准备工作
1 uname -m && cat /etc/*release # get ubuntu version
2 uname -r # get ubuntu kernal version
3 gcc --version # get gcc version
4 lspci | grep -i nvidia # check nvidia GPU information
5 cat /proc/driver/nvidia/version # check nvidia GPU driver version
在https://developer.nvidia.com/cuda-downloads下载了对应的安装包。
然后执行安装命令:
过了一小段时间,大约几分钟吧,安装完成:
之后,执行如下命令:
apt-get update
apt-get install cuda
安装完成后,执行如下命令:
sudo vi ~/.bashrc # add PATh and LD_LIBRARY_PATH at the end
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
Source ~/.bashrc
sudo reboot
重启后尝试:
dpkg -l | grep nvidia
nvidia-smi
cd /usr/local/cuda-8.0/samples
make # compile cuda-8.0 sample files
cd bin/x86_64/linux/release/ # run sample file
./deviceQuery
./bandwidthTest
到此升级cuda到8.0结束。