ubutnu 18.04 安裝tensorflow-gpu-2.1

參考的是官方教程，python版本是3.7.6

（1）安裝tensorflow

    pip install tensorflow  # stable

（2）安裝GPU相關

    # Add NVIDIA package repositories
    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
    sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-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

    # Install NVIDIA driver
    sudo apt-get install --no-install-recommends nvidia-driver-418
    # Reboot. Check that GPUs are visible using the command: nvidia-smi

    # Install development and runtime libraries (~4GB)
    sudo apt-get install --no-install-recommends \
        cuda-10-1 \
        libcudnn7=7.6.4.38-1+cuda10.1  \
        libcudnn7-dev=7.6.4.38-1+cuda10.1
    

    # Install TensorRT. Requires that libcudnn7 is installed above.
    sudo apt-get install -y --no-install-recommends libnvinfer6=6.0.1-1+cuda10.1 \
        libnvinfer-dev=6.0.1-1+cuda10.1 \
        libnvinfer-plugin6=6.0.1-1+cuda10.1
 

       

重啓以後，輸入用戶名密碼然後黑屏，再重啓看不到GRUB界面，繼續讓登錄，死循環。。。

在網上搜索後，讓開機後常按shift鍵，我的電腦長按shift鍵還是沒能看到GRUB界面

然後按照網上說的重啓後按ESC，終於進到GRUB，這塊試着按幾次找找感覺，應該是可以進來的

然後選擇第二個選項：高級選項，第一項是ubuntu（也就是進入ubuntu系統）

然後接着選第二項，recovery mode

然後卸載英偉達驅動

sudo apt-get purge nvidia*

這裏我順便整理了下源列表，把軟件升級了一下

源包括兩個部分/etc/apt/ 下面的source.list和source.list.d文件夾

整理好源列表後

sudo apt-get update
sudo apt-get upgrade

然後重啓電腦登陸順利進入桌面

再然後用電腦自帶的軟件和更新功能安裝驅動

圖片是從網上找的，我裝驅動版本是435，沒安裝之前驅動應該是最下面那個選項

 

驅動裝好後，重啓電腦

然後命令行窗口輸入

nvidia-smi

成功看到驅動和CUDA版本號

然後測試Pytorch和tensorflow都可以正常使用。

大工告成，花了好幾個小時

收穫還是有的，對Ubuntu系統的修復又多了一些瞭解，看到黑屏不會那個緊張了