Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置

我正在记录Ubuntu16.04配置  pytorch+Nvida显卡+cuda8.0的安装过程。


目前环境:

系统: Ubuntu 16.04 LTS

Anacond3—5.0.1

conda4.3.30

Python 3.6.3 :: Anaconda custom (64-bit)

显卡:GeForce GT 720M/PCIe/SSE2

已经安装英伟达。

英伟达驱动: Nvidia Driver Version   340.102

没有配置cuda8.0版本。


1.Nvida显卡安装方法

第一步:进入“Setting—Software&Update”

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第1张图片
Setting—Software&Update

第二步:切换到Additional Drivers选项卡,

        然后选择上面的,然后再点击右下角的Apply Chang,等待一会就好了,然后按照提示“重新启动”,安装完成。

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第2张图片
Additional Drivers界面

第三步:测试/查看是否安装成功

方法一:在终端中输入以下命令,如果出现版本号就说明安装成功

lspci | grep -i nvidia


测试&安装成功效果图

方法二:直接按 开始键 进入查找程序的界面,输入 n ,就可以看到Nvidia的小图标了^_^

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第3张图片
Nvidia的小图标,请忽略后面的网易云音乐
Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第4张图片
这是Nvidia大图标(参数也可以看到o~)

2.CUDA8.0安装

第一步:进到CUDA Toolkit下载网站

请点击:CUDA Toolkit下载页面

按照系统参数进行配置,我的电脑配置如下:然后点击右下角的Download按钮

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第5张图片
CUDA Tooklkit下载选择界面

第二步:按照官网的给的命令教程继续安装

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第6张图片
官网给出的教程

注意:在进行第2条时,其中的cuda-repo-需要换成cuda-repo-9-0-local  (在上面用红框标出了已经),剩下就是等待了。。。

第三步:测试/查看安装是否成功。

方法一:在上面第二步进行完之后,会看到终端里显示两个done,说明就成功了

第二步等待之后,不在终端出现done,done的界面

方法二:与安装Nvidia显卡测试一样,在按了 开始键 之后,输入 cuda 就会出现 两个漂亮的小   图标了

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第7张图片
漂亮的CUDA图标

3. 安装Pytorch的详细过程

方法一:直接去pytorch官网 www.pytorch.org 安装就好了。

方法二(我现在用的):在Anaconda Cloud里搜索 关键字pytorch和torchvision 进行安装,界面就不演示了,直接上安装命令行

第一条命令: 安装pytorch。

conda install -c peterjc123 pytorch

应该会安装下图中的这些包(可以看到安装的pytorch是0.1.12版,最新的是0.2.1版)

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第8张图片
输入 conda install -c peterjc123 pytorch后。。。

第二条命令:安装torchvision

conda install -c soumith torchvision

应该会安装下图中的这些包(这里会更新pytorch到0.2.1版 ^_^)

Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置_第9张图片
输入 conda install -c soumith torchvision

第三步:测试Pytorch是否安装成功

在终端,输入第一条命令 :python

然后输入第二条命令:import torch

如果没有报错,就成功了!

测试Pytorch是否安装成功

FINE!

你可能感兴趣的:(Ubuntu 16.04 + Anaconda3-5.01 + Pytorch0.2.1 + Cuda8.0 + cuDNN6.0.1安装配置)