我正在记录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”
第二步:切换到Additional Drivers选项卡,
然后选择上面的,然后再点击右下角的Apply Chang,等待一会就好了,然后按照提示“重新启动”,安装完成。
第三步:测试/查看是否安装成功
方法一:在终端中输入以下命令,如果出现版本号就说明安装成功
lspci | grep -i nvidia
方法二:直接按 开始键 进入查找程序的界面,输入 n ,就可以看到Nvidia的小图标了^_^
2.CUDA8.0安装
第一步:进到CUDA Toolkit下载网站
按照系统参数进行配置,我的电脑配置如下:然后点击右下角的Download按钮
第二步:按照官网的给的命令教程继续安装
注意:在进行第2条时,其中的cuda-repo-<version>需要换成cuda-repo-9-0-local (在上面用红框标出了已经),剩下就是等待了。。。
第三步:测试/查看安装是否成功。
方法一:在上面第二步进行完之后,会看到终端里显示两个done,说明就成功了
方法二:与安装Nvidia显卡测试一样,在按了 开始键 之后,输入 cuda 就会出现 两个漂亮的小 图标了
3. 安装Pytorch的详细过程
方法一:直接去pytorch官网 www.pytorch.org 安装就好了。
方法二(我现在用的):在Anaconda Cloud里搜索 关键字pytorch和torchvision 进行安装,界面就不演示了,直接上安装命令行
第一条命令: 安装pytorch。
conda install -c peterjc123 pytorch
应该会安装下图中的这些包(可以看到安装的pytorch是0.1.12版,最新的是0.2.1版)
第二条命令:安装torchvision
conda install -c soumith torchvision
应该会安装下图中的这些包(这里会更新pytorch到0.2.1版 ^_^)
第三步:测试Pytorch是否安装成功
在终端,输入第一条命令 :python
然后输入第二条命令:import torch
如果没有报错,就成功了!