本人的ubutu版本為16.04，cuda9.0。

一.啟動ubuntu系統出現黑畫面
1.開機，進入grub畫面。選擇第一項”ubuntu”，按”e”，進入編輯模式。將”quite splash”, 修改為”quite splash nomodeset”。
2.按 ”F10”啟動系統。
3.進入系統後編輯 “/etc/default/grub”

    sudo gedit /etc/default/grub
找到這一行: 
GRUB_CMDLINE_LINUX_DEFAULT=”quiet splash” 
修改為： 
GRUB_CMDLINE_LINUX_DEFAULT=”quiet splash nomodeset”
 

4.儲存

5.更新GRUBsudo update-grub

6.重啟

二. 安裝 CUDA Toolkit

1.官方下載地址：https://developer.nvidia.com/cuda-toolkit 調節nomodeset之後，驅動程式丟失會導致Linux卡頓，這時安裝好CUDA就可以解決所有問題。 （CUDA預設包含顯示卡驅動，所以不需要額外安裝顯示卡驅動）

    sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
    sudo apt-get update
    sudo apt-get install -y cuda
 

2. 安裝 nVidia cuDNN library (cudnn-9.0-linux-x64-v7.1)

    tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz
    sudo cp cuda/include/cudnn.h /usr/local/cuda/include
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

三.安裝Anaconda for python 3.6

    bash ~/Downloads/Anaconda3-5.1.0-Linux-x86_64.sh 
    export  PATH=/home/sll2/anaconda3/bin:$PATH  （注意這裡根據自己的路徑進行書寫）
 

然後,輸入python檢視自己的python版本多少。

四、安裝pytorch

conda install pytorch torchvision cuda90 -c pytorch

驗證pytorch是否安裝成功，在終端輸入：python，進入到python環境，繼續輸入：

import torch
import torchvision

不報錯就表明安裝成功。

注意需要先安裝好git，沒安裝的使用apt install git命令進行安裝。然後執行下面兩條命令：

pip install git+https://github.com/pytorch/[email protected]
pip install --upgrade git+https://github.com/pytorch/[email protected]