Ubuntu16+pytorch -GPU环境配置（超详细步骤，有图）

 

1、安装 Cuda8.0和cudnn7.0.5

Caffe安装教程中已经讲过了，见ubuntu16.04+caffe-GPU环境配置.docx。

2、安装anaconda3.5.2

下载网站https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/，选择Anaconda3-5.2.0-Linux- x86 _64.sh，然后安装：

• sudo sh Anaconda3-5.2.0-Linux- x86 _64.sh

3、Anaconda安装 Pytorch & torchvision

1)获取安装命令

首先进入 Pytorch 官网： https://pytorch.org/

依次选择你电脑的配置。例如，我在Linux下用 Anaconda 3.5.2 安装，Anaconda 3.5.2 默认安装了 python 3.6，GPU (即CUDA-8.0)，如图3就选择了如下设置：

图3 Pytorch官网获取命令

 

2) Pytorch0.4.1安装

在命令行中输入安装命令：

• conda install pytorch torchvision cuda80 -c pytorch

输入 y ，回车。一般情况下，会顺利安装完成。但我遇到了如下问题：

An HTTP error occurred when trying to retrieve this URL. ............

遇到这种情况，我查了下，大概就是网络问题，重新运行一次，就成功了。

3)测试Pytorch是否安装成功

• python
• import torch

如图6所示，没报错，即安装成功。

4、Anaconda安装Dlib19.5.0

• pip install dlib==19.5.0

5、Anaconda安装opencv3.4.3

• pip install opencv-python

6、部分库安装

• conda install numpy
• conda install scipy
• conda install matplotlib