上周末由于某些莫名的原因重装了系统,又重新捣鼓了一次深度学习环境全套安装~~
吐血~~做个记录,方便下次继续重装系统又要安装。。。。。
本次所有安装基于Ubuntu16.04系统下安装,安装好后的环境是
nvidia-410,
Cuda10.0.130,
Cudnn7.6.5,
Anconda5.1(Python3.6.4),
Tensorrt7.0.0.11。
安装NVIDIA驱动
卸载干净所有安装过的nvidia驱动
sudo apt-get remove --purge nvidia-*
执行以下命令添加驱动源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
Ctrl+Alt+F1切换到tty1执行
sudo service lightdm stop
sudo apt-get install nvidia-410 nvidia-settings nvidia-prime
sudo nvidia-xconfig
sudo update-initramfs -u
sudo reboot
上面sudo apt-get install nvidia-410 nvidia-settings nvidia-prime
这条语句表示我安装的nvidia驱动版本是410,cuda和显卡版本对应关系如下:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
会花费很长很长的时间,,,我挂了一晚上去装。。。看网速看运气吧。。。
安装CUDA10.0
点击cuda官网下载链接
下载对应版本的deb安装包,安装说明安装即可。
cd到cuda安装包路径下,执行:
sudo dpkg -i cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
以上,完成cuda10.0安装
安装cudnn7.6.5
点击官网下载链接下载对应的tar包,
cd到cuda tar包下面,解压tar包,
执行:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
然后
gedit ~/.bashrc
在最后面添加如下内容即可
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
以上,完成cudnn的安装。
检验cuda和cudnn是否安装成功如下:
# 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
# 查看cuda版本
cat /usr/local/cuda/version.txt
安装Anaconda
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
过于简单忽略。。。
安装Pycharm
下载Pycharm tar包,解压,
添加快捷方式方法
第一:
sudo gedit /usr/share/applications/pycharm.desktop
第二:在里面添加内容
[Desktop Entry]
Type=Application
Name=Pycharm
GenericName=Pycharm3
Comment=Pycharm3:The Python IDE
Exec="/home/delian/tools/pycharm-community-2019.3.1/bin/pycharm.sh" %f
Icon=/home/delian/tools/pycharm-community-2019.3.1/bin/pycharm.png
Terminal=pycharm
Categories=Pycharm;
注意Exec
和Icon
的路径
添加可执行权限:
sudo chmod +x /usr/share/applications/pycharm.desktop
OK!
安装Tensorrt
- 下载tensorrt安装包(deb)
- 按照安装说明安装
- 安装pycuda
终端执行:
source activate python环境
pip install 'pycuda>=2019.1.1'