在GTX1080上运行FPN-pytorch

环境介绍

  torch版本：0.4.1
  Python版本：3.6
  torchvision版本：0.2.1
  需要tqdm、tensorboardX
  Ubuntu16.04（这是我自己的系统版本）

项目地址

  easy-fpn.pytorch

服务器环境验证

  验证环境是否符合要求输入以下命令

lsb_release -a #查看系统版本
$ python -V #查看python版本
$ python -v
>>import torch
>>print(torch.__version__) # 查看torch版本
>>print(torch.version.cuda) #查看torch里cuda编译版本
>>exit #退出
$ nvcc -V #查看cuda版本
$ nvidia-smi #查看GPU情况

  如果符合就继续，如果不符合就修改环境，修改环境的操作这里就不细说了。下面说一说步骤。

一、克隆项目

我们先将easy-fpn.pytorch的仓库克隆到你想要存放的文件夹下，比如我这里是pycharmprojects，输入以下命令：

$ cd pycharmprojects
$ git clone --recursive https://github.com/potterhsu/easy-fpn.pytorch.git

  --recursive用于循环递归克隆子项目

二、创建数据文件夹

$ cd easy-fpn.pytroch
$ mkdir data

这个没啥好说的

三、下载数据

• 下载压缩文件到data

$ wget http