在矩池云上复现 CVPR 2018 LearningToCompare

这是 CVPR 2018 的一篇少样本学习论文：Learning to Compare: Relation Network for Few-Shot Learning

源码地址：https://github.com/floodsung/LearningToCompare_FSL

环境选用 Tensorflow 1.4 因为他是 cuda8 的。

切换conda源

bash /public/script/switch_conda_source.sh

创建虚拟python环境

conda create -n py27 python=2.7

conda deactivate
conda activate py27

安装 torch 0.3

接下来的任务是找 torch 0.3 的whl安装包，我从下面的链接中找到了

https://download.pytorch.org/whl/cu80/torch_stable.html

我这里是直接pip，复制下面的命令即可。

pip install https://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
pip install torchvision==0.2.1
pip install matplotlib scipy

pip list

拉取github库

git clone https://github.com/brendenlake/omniglot.git

我这里用了一个github镜像来完成

git clone https://hub.fastgit.org/floodsung/LearningToCompare_FSL.git
cd LearningToCompare_FSL/
ls

解压文件并测试运行

cd /LearningToCompare_FSL/datas
unzip omniglot_28x28.zip
cd /LearningToCompare_FSL/omniglot
python omniglot_train_one_shot.py -w 5 -s 1 -b 19

查看有没有使用到gpu

nvidia-smi -l 5

查看文章

矩池云上如何加速 GitHub 下载？

矩池云上执行 conda install 的时候下载特别慢怎么办？怎么切换源？

https://pytorch.org/get-started/previous-versions/

在矩池云上复现 CVPR 2018 LearningToCompare_FSL 环境相关推荐

矩池云上复现论文 Neural Graph Collaborative Filtering 环境复现
矩池云上复现论文 Neural Graph Collaborative Filtering 环境复现 Neural Graph Collaborative Filtering (NGCF) is a ...
在矩池云上复现 PaddleGAN 照片转油画风格教程
我选用的是cuda10.1纯镜像案例来自paddlepaddle官方PaddleGAN库中,可以查看https://github.com/PaddlePaddle/PaddleGAN/ 安装需要的软 ...
矩池云上创建Pytorch 0.41环境
换源 sh /public/script/switch_conda_source.sh 创建python3.6虚拟环境 conda create -n py36 python=3.6.2 进入3.6虚 ...
如何在矩池云上安装语音识别模型 Whisper
如何在矩池云上安装语音识别模型 Whisper Whisper 是 OpenAI 近期开源的一个语音识别的模型,研究人员基于 680,000 小时的标记音频数据进行训练,它同时也是一个多任务模型,可以 ...
矩池云上CUDA、nvcc、cuDNN相关操作汇总
本文汇总整理如何在矩池云上安装.使用CUDA .nvcc.cudnn的教程,后续还将不断补充,也欢迎大家在评论区提相关问题. 优质提问和反馈,将有机会获得矩池云算力豆激励- CUDA相关文章 CUDA ...
矩池云上安装CUDA头文件教程
简介 CUDA头文件(cuda header files)在进行CUDA编程的时候会经常遇到.下面我介绍了一些常用的CUDA头文件及其安装. 英伟达官方源矩池云cuda10及以上 https://d ...
矩池云上使用nvidia-smi命令教程
简介 nvidia-smi全称是NVIDIA System Management Interface ,它是一个基于NVIDIA Management Library(NVML)构建的命令行实用工具, ...
如何在矩池云上安装java
在矩池云上安装java其实非常简单,按照下面的办法安装即可安装 OpenJDK 11 apt update apt install openjdk-11-jdk java -version 安装 O ...
在矩池云上使用A6000/3090跑ikatago说明
因为英伟达30系和a系列对cuda版本的要求是cuda11及以上. https://github.com/kinfkong/ikatago-resources/tree/master/dockerfi ...

在矩池云上复现 CVPR 2018 LearningToCompare_FSL 环境