ubuntu安装cuda8.0+tensorflow+pytorch

Data: 2017.7.23

Author: cjh

1. 下载cuda8.0

本人选择的是deb(local)，网上很多教程都是根据runfile(local)安装的，但是我使用该方法总是安装失败

2. 安装cuda

下载完成后，将其文件放到/home/user目录

sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

完成安装后执行nvcc –version

ubuntu安装cuda8.0+tensorflow+pytorch

出现cuda版本信息则说明安装成功

否则可根据相关提示完成安装

3. 添加cudnn库

此步骤我也不懂什么意思，网上的教程都这么干

ubuntu安装cuda8.0+tensorflow+pytorch

此处安装是的cudnn v5.1 Library for linux

下载完成后同样放到/home/user目录下

tar xvzfcudnn-8.0-linux-x64-v5.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*

4. 添加环境变量

sudo vim ~/.barchrc

在最后面添加环境变量

export PATH=/usr/local/cuda-8.0/bin:$PATH

exportLD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

source ~/.barchrc

5. Tensorflow安装

网上有很多tensorflow的安装教程，而我选择的是下载好本地安装的方法

在该网址中可找到，对应python版本带gpu或不带的tensorflow版本

下载完成后至于用户目录

sudo pip install tensorflow(根据下载的实际版本做修改)

直接下载往往会由于网络原因，导致失败，具体原因你应该懂

6. Pytorch安装

同样现在pc上下载好对应的pytorch版本

下载完成后至于用户目录

sudo pip install pytorch(根据下载的实际版本做修改)

sudo pip install torchvision