1、安装ubuntu16.04
建议安装桌面版,我安装服务器版时在安装界面会卡住
ubuntu-18.04.2-desktop-amd64.iso
安装完后
sudo passwd更改root密码
重装vim完整版
apt remove vim-common
apt install vim
安装显卡驱动时,关闭图形界面
systemctl set-default multi-user.target
service lightdm stop
可在需要时打开图形界面
systemctl set-default graphical.target
service lightdm start
2、禁止集成的nouveau驱动
ls -lh /etc/modprobe.d/blacklist.conf
chmod 666 /etc/modprobe.d/blacklist.conf
vi /etc/modprobe.d/blacklist.conf
在blacklist.conf最后添加
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb
3、安装显卡驱动
卸载所有安装的nvidia驱动:
apt remove --purge nvidia-*
安装方法1,需要先查询显卡驱动号如440:
add-apt-repository ppa:graphics-drivers/ppa
apt update
apt install -y nvidia-440 mesa-common-dev freeglut3-dev
apt-get install --no-install-recommends nvidia-440 libcuda1-440 nvidia-opencl-icd-440
安装方法2:到官网下载合适版本
https://www.nvidia.com/Download/index.aspx?lang=en-us
安装驱动
./NVIDIA-Linux-x86_64-440.31.run -no-opengl-files -no-opengl-files -no-x-check -no-nouveau-check
最后安装
apt install nvidia-cuda-toolkit
4、BIOS设置,重要
bios关闭secure boot
设置thunderbult secure level为最低legacy
重启
lspci -tv
或nvidia-smi
即可看到显卡
5、安装cuda9.0+,选择不安装显卡驱动
运行cuda_9.0.176_384.81_linux.run
安装完,在~/.bashrc后加上:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
6、安装cudnn9.0
cp cuda/include/cudnn.h /usr/local/cuda/include/
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
chmod a+r /usr/local/cuda/include/cudnn.h
chmod a+r /usr/local/cuda/lib64/libcudnn*
查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
到此CUDA环境安装成功
7、安装PYTHON环境
bash Anaconda3-5.2.0-Linux-x86_64.sh
pip install --no-index --find-links=file:./site-packages -r requirements.txt --ignore-installed