caffe我选择使用ubuntu源码安装,所以先执行:
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev protobuf-compiler libhdf5-serial-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler protobuf-c-compiler python-pandas
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get update
sudo apt-get upgrade
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
#1:装驱动 安装英伟达驱动
sudo apt-get install nvidia-375 (sudo ubuntu-drivers devices 查推荐的驱动)
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
sudo apt-get update
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
从英伟达官网下载对应的驱动然后使用命令安装
sudo dpkg -i libcudnn7_7.3.0.29-1+cuda9.0_amd64.deb
sudo apt-get update
sudo dpkg -i libcudnn7-dev_7.3.0.29-1+cuda9.0_amd64.deb
sudo apt-get update
sudo dpkg -i nccl-repo-ubuntu1604-2.3.4-ga-cuda9.0_1-1_amd64.deb
sudo apt-get update
编辑 /etc/modprobe.d/blacklist-nouveau.conf
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
- 打开后发现该文件中没有任何内容,写入:
blacklist nouveau option nouveau modeset=0
保存后关闭文件,注意此时还需执行以下命令使禁用 nouveau 真正生效:
sudo update-initramfs -u
之后重启系统让GTX1080显卡驱动生效。
终端输入:
nvidia-smi,会出来如下显卡信息。
#安装cuda
按照使用手册来安装
Installation Instructions: | |
|