配置MSI GPU 深度学习环境

安装 Ubuntu 18.04, CUDA, CDNN, Pytorch 和 TensorFlow

安装 Ubuntu 18.04

获取镜像

https://tutorials.ubuntu.com/tutorial/tutorial-create-a-usb-stick-on-ubuntu#0

安装

• 关闭Security boot (可以在BIOS里进行关闭，对于MSI电脑，重启操作系统，在系统载入前，不断按delete键进入BIOS)
• 可能的问题: 安装时，屏幕卡住
  • 重启系统
  • 选择安装Ubuntu的选项(但是不要按ENTER键进入)
  • 按 e
  • 找到以linux开头的一行代码，将modprobe.blacklist=nouveau 添加到 quiet splash 后面.
• 选择Wifi网络，并安装第三方软件以及更新
• 一旦安装完成，移除安装盘并重启系统

安装Nvidia驱动、CUDA、CUDNN

• 在GRUB菜单选择Ubuntu(如果出现屏幕卡住的情况，重复modprobe.blacklist=nouveau步骤).

• 确保security boot关闭

• 进入命令行 (ctrl+alt+ F1,F2,F3…F??)

  # 添加NVIDIA源
  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
  
  sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
  
  sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
  
  sudo apt-get update
  
  wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
  
  sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
  
  sudo apt-get update
  
  # 安装NVIDIA驱动
  sudo apt-get install --no-install-recommends nvidia-driver-418
  
  # 重启。使用nvidia-smi命令查看是否安装成功
  
  # 安装CUDA、CUDNN开发和运行环境(~4GB)
  sudo apt-get install --no-install-recommends \
  cuda-10-0 \
  libcudnn7=7.6.2.24-1+cuda10.0  \
  libcudnn7-dev=7.6.2.24-1+cuda10.0
  
  
  # 安装TensorRT。需要先安装上述的libcudnn7。
  sudo apt-get install -y --no-install-recommends libnvinfer5=5.1.5-1+cuda10.0 \
  libnvinfer-dev=5.1.5-1+cuda10.0
  
  

安装 pyenv

# 安装pyenv
$ git clone https://github.com/pyenv/pyenv.git ~/.pyenv
$ echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc
$ echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc
$ echo -e 'if command -v pyenv 1>/dev/null 2>&1; then\n  eval "$(pyenv init -)"\nfi' >> ~/.bashrc
$ exec "$SHELL"
# 安装python
$ pyenv install 3.7.4

# 安装pyenv-virtualenv 
$ git clone https://github.com/pyenv/pyenv-virtualenv.git ~/.pyenv/plugins/pyenv-virtualenv
$ echo 'eval "$(pyenv virtualenv-init -)"' >> ~/.bashrc
$ exec "$SHELL"

# 创建环境
$ pyenv version
3.7.4 (set by /home/yyuu/.pyenv/version)
$ pyenv virtualenv ML

安装Pytorch1.3和TensorFlow (2.0)

# 激活ML环境
$ pyenv activate ML

# 安装Pytorch
$ pip install torch torchvision

# 安装Tensorflow
$ pip install tensorflow-gpu