ubuntu 1604安装cuda 10.1 ,cudnn 10.1,pytorch 1.3.1 ,libtorch

1、安装cuda10.1

从这里下载cuda10.1的toolkits
cuda 10.1下载链接
按下图选择

安装步骤就按照Base Installer中的四句指令来安装，第二句的指令cuda-repo- 需要修改，在执行弯第一句指令后terminal log的最后一句里，直接复制粘贴到命令行。cuda toolkit会帮忙我们将nvidia驱动也一并安装好。安装完之后在终端窗口中输入：nvidia-smi指令，如果显示如下，则说明cuda已经安装好了。

可能会有用的链接黑屏问题解决 https://blog.csdn.net/qq_39670011/article/details/90404111 ，Ubuntu16.04+cuda10.1官方安装指导的补充说明。这篇帖子中说需要对显卡驱动做一些设置，否则重启电脑可能会黑屏，不过我们的电脑并没有出现所说的问题，暂且将网址记录到这里，如果出现问题可以参考解决。

2、安装cudnn 10.1

下载cudnn 10.1，官网：cudnn下载地址 https://developer.nvidia.com/rdp/cudnn-archive。选择对应cuda 10.1的版本。我选择了最上面的这个

安装方法参考：cudnn安装方法 https://blog.csdn.net/suezhang9/article/details/98657431。这个链接里说了两种安装方法，我按第一种安装方法安装了一下。就是将对应的头文件和库文件拷贝到了指定的目录下。

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

3、安装pytorch

3.1 libtorch

打开官网 https://pytorch.org/，因为我要用C++开发，所以按下图选择，按照”Download here (cxx11 ABI)“的链接下载然后解压到自己指定的目录下。

这是libtorch官方安装使用说明：安装链接 https://pytorch.org/cppdocs/installing.html

使用的时候需要在自己的CMakeLists.txt中添加下面两句cmake指令，然后包含头文件和lib文件

list(APPEND CMAKE_PREFIX_PATH /path/to/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)
include_directories(${TORCH_INCLUDE_DIRS})


add_executable(${PROJECT_NAME} your cpp files)

targe_link_libraries(${PROJECT_NAME} ${TORCH_LIBRARIES})

3.2 pytorch

安装python版的pytorch，按下图选择。我的python版本是3.6。这里简单的按照自己的python版本选择就可以。

使用pip3安装的时候遇到了问题，是pip3的问题，修改一下/usr/bin目录下的pip3脚本如下，并保存。然后就可以使用pip3进行安装了。

然后执行：

sudo pip3 install torch torchvision

安装就结束了。