Ubuntu必备技能——深度学习环境Nvidia、Cuda、Cudnn、Anaconda、Pytorch、Pycharm、Vim、虚拟环境、Tmux分屏复用

深度学习环境配置

一、安装NVIDIA显卡驱动：

1.1屏蔽

sudo vim /etc/modprobe.d/blacklist.conf

—文件最后添加下面几行：

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

更新、重启

sudo update-initramfs -u
sudo reboot

重启后，得等一会才能正常连接服务器，输入下面指令

lsmod | grep nouveau

没有输出表示屏蔽成功

1.2 安装NVidia驱动

sudo apt-get --purge remove nvidia-*
sudo chmod a+x NVIDIA-Linux-x86_64-384.130.run
sudo ./NVIDIA-Linux-x86_64-384.130.run -no-x-check -no-nouveau-check -no-opengl-files

选项全部yes即可

nvidia-smi

显示显卡信息，安装正常

二、安装anaconda

• bash Anaconda3-2020.07-Linux-x86_64.sh
• 选项全部默认，yes，即可
• 最后一定要更新环境变量，source ~/.bashrc
• conda --version
• 安装成功

三、利用conda创建python虚拟环境

• conda create -n your_env_name python=3.6 创建虚拟环境
• conda activate your_env_name 激活虚拟环境
• conda deactivate 关闭当前虚拟环境，切换到base环境
• conda remove -n your_env_name --all 删除虚拟环境
• conda list 查看安装了哪些包
• conda install numpy 安装包
• conda env list 查看当前存在的虚拟环境
• conda update conda 检查更新当前conda

四、配置pytorch深度学习环境

• conda install pytorch=1.6.0 torchvision -c pytorch

• 安装pytorch，可以指定版本，不指定的话默认装最新版的，另外，conda会自动安装版本匹配好的cudatoolkit(cuda)，不过不会自动安装cudnn，但是这个好像不装也没事，反正跑代码的时候可以使用GPU。

• 建议不要在主环境安装cuda，

• 如果在主环境base下安装cuda，就不要在虚拟环境中安装cuda了，否则会版本冲突，同时，虚拟环境安装pytorch要是由pip安装，因为conda install会自动安装cuda

如果在主环境下安装了cuda和cudnn，建议利用pip安装pytorch

• pip install torch==1.6.0
• pip uninstall torch
• 临时更换源：pip install scrapy -i https://pypi.tuna.tsinghua.edu.cn/simple

主环境下安装cuda，cudnn，建议参考下面博客：

Ubuntu18.04安装CUDA10、CUDNN

五、安装pycharm

• 官网下载pycharm的linux安装包
• 解压到/opt/路径下：tar -xzf pycharm-professional-2019.1.3.tar.gz -C /opt/
• 进入解压后的bin目录：cd /opt/pycharm-2019.1.3/bin
• 安装pycharm：sh ./pycharm.sh
• 快捷方式创建：Tools–>Create Desktop Entry

Vim常用指令

• vim filename.py默认进行命令模式
• 按下i，进入输入模式，可以输入字符
• 按下ESC，退出输入模式，切换到命令模式
• 输入:wq保存退出，:q退出不保存，:wq!强制保存退出

基本操作指令

• 创建文件：touch a.txt

• 创建文件夹：mkdir newfolder

• 删除文件：rm a.txt

• 删除文件夹：rmdir newfolder

• 删除文件夹及其中的所有文件：rm -r newfolder

• 删除文件夹下的全部文件：rm -r *

• 复制、移动文件到指定路径：cp/mv [文件名] [路径名]

• 复制、移动文件夹下的所有文件（不包括文件夹）到指定路径：cp/mv [文件夹路径]/* [路径名]

• 复制、移动文件夹下的所有文件（包括文件夹）到指定路径：cp/mv [文件夹路径]/ [路径名]

• 查找文件：sudo find / -name pycharm*

• 查看进程：ps -aux | grep pycharm

• 强制杀死进程：sudo kill -s 9 3758(进程编号)

• 修改环境变量 vim ~/.bashrc

• 生效 source ~/.bashrc

tmux分屏复用

• 打开终端，输入tmux，便进入可切分的终端状态（通过Ctrl+b来激活控制台，这个时候可以通过按键进行操作，每一次的操作，都需要先激活控制台Ctrl+b）
• 左右分屏：Ctrl+b，再按下%键（shift+5）
• 上下分屏：Ctrl+b，再按下"键
• 切换终端：Ctrl+b，再按下o键（英文字母o）
• 关闭当前子终端页面：Ctrl+b，再按下x键（英文字母x）
• 关闭整个页面：Ctrl+b，再按下&键，这个时候会提示你是否关闭，kill-window bash?(y/n)，输入y即可关闭
• 分屏之后，如果当前屏无法显示完整信息，可以通过Ctrl+b进入控制台，然后使用键盘上的PgUp和PgDn上下翻页，按下q退出

（连接服务器跑实验的时候，时间长的时候会自动断开，利用tmux建立会话，可以完美的解决这个问题）

• tmux new -s JpgToJpg 新建一个会话
• ctrl + b 再按下d，脱离当前会话
• tmux new -s RawToJpg 再新建一个会话
• tmux ls 列出会话
• tmux kill-session -t 会话名，关闭会话
• tmux at -t 会话名，恢复会话