pytorch和torchnet的安装与测试（Ubuntu16.04+cuda9.0）

本人的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)

官方下载地址：https://developer.nvidia.com/cudnn

 

    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

Anaconda官网：https://www.anaconda.com/download/

    bash ~/Downloads/Anaconda3-5.1.0-Linux-x86_64.sh 
    export  PATH=/home/sll2/anaconda3/bin:$PATH  （注意这里根据自己的路径进行书写）

然后,输入python查看自己的python版本多少。

四、安装pytorch

根据官网：http://pytorch.org/

conda install pytorch torchvision cuda90 -c pytorch

验证pytorch是否安装成功，在终端输入：python，进入到python环境，继续输入：

import torch
import torchvision

不报错就表明安装成功。

五、安装torchnet 参考https://github.com/pytorch/tnt

注意需要先安装好git，没安装的使用apt install git命令进行安装。然后执行下面两条命令：

pip install git+https://github.com/pytorch/tnt.git@master
pip install --upgrade git+https://github.com/pytorch/tnt.git@master

参考网站：https://github.com/pytorch/tnt

参考博客：https://blog.csdn.net/shiorioxy/article/details/78850199

                       https://blog.csdn.net/hjxu2016/article/details/78527019?locationNum=1&fp