一、电脑配置

说明:
电脑配置:
LEGION笔记本
CPU Inter Core i7 8代
GPU NVIDIA GeForce GTX1060
Windows10

所需的环境:

Anaconda3(64bit)
CUDA-9.0
CuDNN-7.1

二、安装cuda

1.查看自己电脑NVIDIA图形卡是否支持GPU运算

在安装之前你要先查看你的电脑是否支持GPU运算,否则你也不用安装了。

打开终端:

方法一:

ubuntu-drivers devices

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

我的显卡是GTX1060的

方法二:

可以查看自己当前主机的显卡型号以及个数

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

2.查看当前主机是否为64位:

uname -a

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

3.安装推荐的显卡驱动程序(重要)

方法(推荐):

首先打开linux系统(Ubuntu16.04)找到设置,设置中的"软件和更新"部分,然后打开,找到“附加驱动”,打开后安装系统推荐的驱动。

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

然后就是静静等待安装完成就可以了。安装完成后记得重启电脑!重启电脑!重启电脑!

这种方法的好处就是不易出错,配置的话电脑都给你做好了。

 

(选)如果你的电脑误把专用的显卡驱动程序给删掉,或者你上面那个图中都没有东西的话,你可以试试下面这个:(我就是。。。脸黑。。),可以先装回来一个:(这步一般大家不用执行)

sudo apt install nvidia-cuda-toolkit

 

4.下载CUDA9.0
地址下载:https://developer.nvidia.com/cuda-downloads

下面这个是10.0目前官网最新

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

1)若要安装其他版本的到:Legacy Releases,我安装的是9.0

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

 选择相对应的版本(下载基本包以及补丁):

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

 

2)安装自己版本的sudo sh cuda_9.0.176_384.81_linux.run
ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

只有驱动部分是不需要安装的,因为之前已经安装过了,剩下的基本上都是yes默认

3)配置环境变量:

打开配置环境的文件:

sudo gedit ~/.bashrc

在打开的文本末尾加入:

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

在终端输入下面的语句,使更改的环境变量生效:

$ source ~/.bashrc

查看是否安装成功:

nvcc -V
ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

或者:

sudo nvidia-smi

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

或者:依次在终端输入:

cd NVIDIA_CUDA-9.0_Samples/0_Simple/vectorAdd
make
./vectorAdd 

结果出现test passed成功:

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

三、安装cudnn7.4.1

(要安装对应cuda版本的cudnn)

安装cudnn :https://developer.nvidia.com/rdp/cudnn-download
1)linux下载第三个:
ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch
2)然后打开终端,1、cd到文件的下载目录;2、解压cudnn9.0文件;3、复制一些文件;
在终端依次中输入以下命令:

cd  cd /media/你的用户名/TOSHIBA\ EXT/alu/CUDA/ # 进入 cuDNN 安装文件的所在路径
tar xvzf cudnn-9.0-linux-x64-v7.4.1.5-1.tgz # 解压
sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 复制到 include 中
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 复制到 lib64 中
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 讲头文件复制进去 

至此,cuda和cudnn安装结束,接下来就可以安装自己需要的东西了

四、安装anaconda安装+创建新环境

如果不安装anaconda,也可以,可以直接在自己环境下pip安装东西,但是因为anaconda会自带很多依赖包,我们不需要安装,而且它还可以创建多个环境实现py2和py3的切换,我觉得这个很好

1)anaconda:Ubuntu 安装 Anaconda3 详细步骤

我按装的是,可以到清华源下载很快:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

2)安装完成之后,如入安装的是anaconda3,输入python,则默认是python3.5

 ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

查看anaconda版本和conda版本:

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

 打开jupyter notebook

jupyter notebook

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

 

3)如果你不用其他环境时,只用py3.5,可以直接conda或者pip来安装一些东西,这个我就多说了,接下来我主要是写在anaconda下创建多个环境,并在该环境中安装依赖包

直接在该环境下安装pytorch(参照下面的 五、)

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

conda install pytorch torchvision -c pytorc
##########测试#############
import
torch as t x=t.rand(5,3) y=t.rand(5,3) if t.cuda.is_available(): x=x.cuda() y=y.cuda() print(x+y)

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

 

如果要多个环境,还不想太乱的话,可以

接下来可以转到我的另一篇,来配置anaconda的多个环境Anaconda3+python3环境下如何创建python2环境(win+Linux下适用,同一个anaconda下py2/3共存)

 

也可以看这里,我以anaconda3下配置环境py2.7下安装pytorch为例:

(1)首先查看环境(如果你只有一个环境,没有创建环境时,所有都默认在一个root环境下):

conda info -e

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

(2)创建新环境(名字叫py27,环境是python2.7)

conda create -n py27 python=2.7

安装完成后,激活环境:

 source activate py27

退出环境:

source deactivate

(3)在新环境下查看当前安装的依赖包:

source activate py27#激活环境
python --version#查看python版本是不是2.7
conda list#目前环境下的包

(4)因为刚刚creat创造py27环境下,只安装一些必要包,所以还需要安装其他包

conda install anaconda

 五、在新环境下安装pytorch

(1)到pytorch官网https://pytorch.org/找对应版本安装

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

下面的run this command:是提示应该输入的命令行

conda install pytorch torchvision -c pytorch=0.4.1#我安装的是pytorch0.4.1版本

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

(2)检验是否安装成功:

import torch
torch.cuda.is_available()#查看是否可用

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

torch.cuda.get_device_name(0)#查看自己的显卡设备

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

x = torch.rand(5,3)
y = torch.rand(5,3)
if torch.cuda.is_available():
       x = x.cuda()
       y = y.cuda()
       print x+y

ubuntun16.04+cuda9.0+cudnn7+anaconda3+pytorch+anaconda3下py2安装pytorch

测试结束。

 参考:https://blog.csdn.net/zzc15806/article/details/81669591