PyTorch——Ubuntu上Pytorch的安装教程

1 致谢

感谢网友夜幕下的光123提供的帮助，
原文链接如下：
https://blog.csdn.net/songchunxiao1991/article/details/95192063

2 前言

今天在学习张老师发的代码～
张老师给的MTCNN代码是用PyTorch实现的，所以我们需要在Ubuntu系统上安装PyTorch～

3 Ubuntu上Pytorch的安装教程

3.1 安装CUDA

step 1： 检查PyTorch库require的CUDA版本，直接在官网进行查看，
例如：PyTorch-1.6支持的最高版本的CUDA是CUDA-10.2；
安装CUDA教程请参考《CUDA——Ubuntu系统上CUDA和cuDNN的安装教程》
使用nvcc -V来确认CUDA版本；

3.2 使用conda来安装PyTorch

step 1： 进入进行安装的conda环境，如：
conda activate conda-general
step 2：
进入PyTorch官网按照说明进行安装，PyTorch-GET-STARTED
step 3： （如果使用官方命令无法正常安装的话，）
就需要使用命令
conda install pytorch torchvision cudatoolkit=10.2
来进行安装，其实官网给出的命令是这样的
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
不过在实际安装的时候会发现安装不成功，会出现“CondaHTTPError: HTTP 000 CONNECTION FAILED for url ……”的错误，which may be due to the existence of BFW, 所以还需要问一下度娘～
后来就找到了网友夜幕下的光123的博文《Pytorch安装–报错-CondaHTTPError: HTTP 000 CONNECTION FAILED for url》，其中说到如果使用官方的命令“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”进行安装，则不会从清华源进行下载，其解决办法就是去掉“-c pytorch”；

3.3 错误解决方案

1、下载速度太慢

1. 使用conda方式安装可能时间较长，也许需要一个时段左右的时间；
2. 使用pip方式安装，可以看到下载速度，比较友好；
3. Try scrambling over the wall.

2、出现“段错误 (核心已转储)”——“之前由于网速太慢中断过安装”
需要对之前下载的临时文件进行清理，使用下面的代码
conda clean -a