环境配置步骤(本文为GPU版)
安装NVIDIA显卡驱动(驱动在官网下载)
1.安装步骤
i) `dpkg -i nvidia-diag-driver-local-repo-ubuntu1604-390.30_1.0-1_amd64.deb’ for Ubuntuii) `apt-get update`
iii) `apt-get install cuda-drivers`
iv) `reboot`
2.测试驱动安装是否成功检查
nvidia-smi
安装cuda
1.安装步骤- `sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb`
- `sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
- `sudo apt-get update`
- `sudo apt-get install cuda`
2.配置环境变量
在.bashrc中加入
$ export PATH=/usr/local/cuda-9.1/bin:$PATH
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.1/l