使用docker跑通tasn-mxnet代码

论文：《Looking for the Devil in the Details: Learning Trilinear Attention Sampling Network for Fine-grained Image Recognition》

代码：TASN

拉取镜像

根据tasn项目需求，从docker官网拉取mxnet/python、cuda8.0的镜像

docker pull mxnet/python:1.5.0_gpu_cu80_mkl_py3

挂载主机

挂载主机使得容器内部目录和宿主主机目录关联起来

sudo docker run --gpus all -it -v 宿主机目录:容器内的目录 镜像id bash

镜像id可通过docker images查看

下载TASN

1、下载tasn：

git clone https://github.com/researchmm/tasn.git
cd tasn/tasn-mxnet/example/tasn

2、安装libopenblas-dev、liblapack-dev和libopencv-dev，建立硬链接：

apt-get install -y libopenblas-dev liblapack-dev libopencv-dev
ln /dev/null /dev/raw1394

配置nccl

1、从谷歌云盘下载nccl.tar

2、将nccl.tar移动到tasn/tasn-mxnet/example/tasn目录下

3、解压nccl.tar：

tar -xvf nccl.tar

4、构建库并测试：

cd nccl
make CUDA_HOME=<cuda install path> test
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./build/lib
./build/test/single/all_reduce_test

如果出现Error: must specify at least data size in bytes!，执行以下命令：

./build/test/single/all_reduce_test 10000000

如果出现以下内容，那么恭喜你

使用docker跑通tasn-mxnet代码相关推荐

YOLOV5使用(一): docker跑通，详解TensorRT下plugin的onnx
yolov5的工程使用(以人员检测为案例) 使用ubuntu为案例 docker run --gpus all -it -p 6007:6006 -p 8889:8888 --name my_torc ...
colab跑通YOLOV3SPP（代码来自霹雳巴拉WZ）
记录以下,方便日后学习 1)下载码源 !git clone https://github.com/WZMIAOMIAO/deep-learning-for-image-processing.git 2 ...
111个Python数据分析实战项目，代码已跑通，数据可下载
写在前面: 这里整理了111个数据分析的案例,每一个都进行了严格的筛选,筛选标准如下: 1. 有干货:杜绝纯可视化.统计性分析,有一定比例的讲解性文字 2. 可跑通:所有代码均经过测试,(大概率)可以 ...
13个月才跑通GitHub模型，机器学习科研入门太难了吧
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达金磊萧箫发自凹非寺量子位报道 | 公众号 QbitAI & ...
win10 安装并跑通 mmdetection
前言:实际项目中还是建议用服务器以及好的显卡.这里我是在自己电脑上(Windows10)一些简单调的实验,之前一直在Linux上跑,也不是第一次了,但这次用windows跑mmdetection还是踩 ...
本科生搞AI在线崩溃：3个月才跑通GitHub模型，机器学习科研入门太难了吧
金磊萧箫发自凹非寺量子位报道 | 公众号 QbitAI "机器学习,太难了!" "3个月才跑通GitHub模型代码,要崩溃了!" "还要跟博 ...
YOLO学习01（跑通yolov5尝试）
目录一.前期理论学习二.使用YOLO(复现yolov5) 1.环境搭建 2.认识YOLO代码中的文件并简单运行(detect.py) 3.模型训练(train.py) 一.前期理论学习绘制思维导 ...
linux docker 跑代码_使用Docker 搭建自动化测试环境
01 我经常是在 windows 上进行代码编写,但是无奈代码写完了总是要放到服务器上跑,自动化测试的代码也不例外. 但是当我把 windows 上写好的代码放到 linux 上的时候,总是会出现各种 ...
yolov3网络结构图_目标检测——YOLO V3简介及代码注释（附github代码——已跑通）...
GitHub: liuyuemaicha/PyTorch-YOLOv3github.com 注:该代码fork自eriklindernoren/PyTorch-YOLOv3,该代码相比master分 ...