HF-NET环境配置与安装

相关链接

论文
github开源项目

环境配置

官方推荐配置：

• Python 3.6
• TensorFlow 1.12 GPU
• PyTorch 0.4.1

安装python3.6

0.ubuntu20.04由于python默认3.8需要改一下
千万不要直接卸载python否则终端就没了，据说ubuntu的可视化界面终端是用python接口的
添加source后按照python3.6

sudo apt install software-properties-common
sudo add-apt-repository ppa:deadsnakes/ppa
sudo apt update
sudo apt install python3.6

使用软连接的方式更改python默认版本

sudo rm /usr/bin/python
sudo ln -s /usr/bin/python3.6 /usr/bin/python

不过创建./venv时候也是用python而不是python3

python -m venv --system-site-packages ./venv

1. 安装python3.6

sudo apt-get update
sudo apt-get install python3.6
sudo apt install python3.6-dev python3-pip python3.6-venv

安装cuda

由于存在版本对应的问题，内容有点多，重新整理了一份参考文章关于多版本的cuda安装切换的。

安装TensorFlow

参考官网推荐的虚拟环境安装
Python 虚拟环境用于将软件包安装与系统隔离开来。

1. 创建一个新的虚拟环境，方法是选择 Python 解释器并创建一个 ./venv 目录来存放它：
   （这里我直接放在了主目录下/home,有需要的可以先cd到指定目录）

python3 -m venv --system-site-packages ./venv

2. 使用特定于 shell 的命令激活该虚拟环境：
   一般ubuntu用的是bash，使用下面一行就可以了

source ./venv/bin/activate  # sh, bash, or zsh

当虚拟环境处于有效状态时，shell 提示符带有 (venv) 前缀。

3. 在不影响主机系统设置的情况下，在虚拟环境中安装软件包。首先升级 pip：

pip install --upgrade pip
pip list  # show packages installed within the virtual environment

4. 安装 TensorFlow pip 软件包

#pip install --upgrade tensorflow-gpu==1.12
#建议换成国内镜像，之前没换下了半天报错timeout？？！！
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.12

5. 验证安装效果：

python3
import tensorflow as tf
tf.test.is_gpu_available()

6. 之后退出虚拟环境：

deactivate  # don't exit until you're done using TensorFlow

安装pytorch

1. 使用特定于 shell 的命令激活该虚拟环境：

source ./venv/bin/activate 

2. 安装torch0.4.1参考官网的cuda对应版本选择cuda9.0https://download.pytorch.org/whl/cu90/torch_stable.html的下载地址

pip install https://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip install torchvision==0.2.1

安装完环境重启一下

之前没重启导致

import tensorflow 
tensorflow.test.is_gpu_available() #返回true，就可以了
import torch
torch.cuda.is_available() #之前没重启，返回false，重启后就正常了

HF-NET安装与demo运行

1. 下载

git clone https://github.com/ethz-asl/hfnet.git

2. 安装

• 因为tensorflow和pytorch虚拟环境安装所有需要先进入环境才能找到之前安装的环境

source ./venv/bin/activate
cd hfnet
make install  #安装时候需要设置两个目录 数据集 实验数据

install过程中需要输入相关的路径信息

Path of the directory where datasets are stored and read: datasets
Path of the directory where experiments data (logs, checkpoints, configs) are written: experiments

运行demo

1.用jupyter打开

jupyter notebook

问题：百度网盘占用端口
解决：直接卸载baidunetdisk

sudo apt-get --purge remove baidunetdisk

2.打开demo.ipynb点击运行