ubuntu安装cuda8.0+tensorflow+pytorch

Data: 2017.7.23

Author: cjh

 

 

1.      下载cuda8.0

https://developer.nvidia.com/cuda-downloads

本人选择的是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

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

 

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

 

3.      添加cudnn库

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

https://developer.nvidia.com/rdp/cudnn-download

此处安装是的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的安装教程，而我选择的是下载好本地安装的方法

https://www.tensorflow.org/install/install_linux#InstallingNativePip

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

下载完成后至于用户目录

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

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

 

6.      Pytorch安装

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

http://pytorch.org/

下载完成后至于用户目录

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

sudo pip install torchvision