因为现在需要安装caffe2,但系统是ubuntu18.08,且要保证tf1.8-gpu健在。各种跳坑后总结出以下规律:
1. ubuntu16.04 + driver 384 + cuda9 + cudnn7 + gcc(<6) : 适用于 tf-gpu 和 caffe2
2. ubuntu18.04 + driver 396 + cuda9.2 + cudnn7.1.4 + gcc(>6) : 适用于 caffe2,是否适用于tf-gpu?
3. tf 对 gcc 没要求,caffe 对 gcc 有要求
以上推论参考:Ubuntu 18.04 + CUDA 9.2 + cuDNN 7.1.4 + Caffe2 安装,tensorflow-gpu1.8.0+win10+vs2017+CUDA9.2 安装
------------------------------------------------------------------------------
所以,现在要尝试上述第2点。首先,Ubuntu18.04 + CUDA9.0 + cuDNN7.1.4 + TensorFlow1.8-Anaconda 的安装,是我的另一篇博客。下面,在该博客的基础上,还需:
a. 更新 CUDA 至 9.2
b. 重装tf
c. 安装caffe2
------------------------------------------------------------------------------
现有驱动:Nvidia 384
需要驱动:Nvidia 396
sudo apt-get purge nvidia*
sudo apt-cache search nvidia-*
sudo apt-get update
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-driver-396
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
sudo reboot #或者sudo shutdown -r now
nvidia-smi
----------------------------------------------------------------
sudo apt-get install gcc-7
sudo apt-get install g++-7
cd /usr/bin
ls -l gcc*
ls -l g++*
sudo mv gcc gcc.bak #备份
sudo ln -s gcc-7 gcc #重新链接
sudo mv g++ g++.bak
sudo ln -s g++-7 g++
gcc -v
g++ -v
出现上图,说明升级成功。
--------------------------------------------------------------
sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt-get remove cuda*
cd /usr/local/
sudo rm -r cuda-9.0
值得注意的是,此时我们刚才安装的驱动还在,不要担心。
CUDA 到 CUDA9.2下载页面下载runfile(安装包 和 1个补丁包)
1. cd到安装包目录,输入以下指令:
sh cuda_9.2.148_396.37_linux.run --override
sh cuda_9.2.148.1_linux.run
2. 出现说明之后,按Ctrl + C键跳过
3. 基本上输入 yes,除了:
在提示是否安装OpenGL时,应该选no。
问你是否需要安装驱动,选择no
sudo gedit ~/.bashrc
export PATH=/usr/local/cuda-9.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64$LD_LIBRARY_PATH
sudo reboot #或者sudo shutdown -r now
1. 终端输入以下,查看cuda版本
nvcc -V
2. 编译samples例子
a. 进入到Samples安装目录,然后在该目录下终端输入make,等待十来分钟:
cd NVIDIA_CUDA-9.2_Samples
make
b. 在Samples里面找到bin/x86_64/linux/release/目录,并切换到该目录, 然后运行deviceQuery程序 :
cd bin/x86_64/linux/release/
sudo ./deviceQuery
c. 如果result = PASS,则安装成功 (如下图),否则失败,可能原因是环境未配置好,或者是cuda与驱动版本不兼容。
----------------------------------------------------------------------
conda install -c caffe2 caffe2-cuda9.0-cudnn7
python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'