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