Ubuntu虚拟环境安装apex包

一个项目需要用到apex包，记录一下安装方式，需要用到的pytorch版本是

pytorch==1.5.1 torchvision==0.6.1

1. 虚拟环境没有NVCC，先安装NVCC

conda install -c conda-forge cudatoolkit-dev=10.1  

2. 虚拟环境安装pytorch

conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch

3. gcc版本降级，应为5.X，偷个懒，借鉴ubuntu 18.0.4 降级安装gcc 5 - 简书 (jianshu.com)，文章中项目使用的ubuntu版本是 18.04, 默认安装的gcc版本是7.5.0

通过apt包管理器进行安装

• sources.list 中添加软件源地址 deb http://cz.archive.ubuntu.com/ubuntu bionic main universe 其中cz.archive.ubuntu.com/ubuntu 可换成国内的镜像源

• 之后update，后重新instll

apt-get update
apt-get install -y gcc-5
apt-get install -y g++-5

安装完成后，系统中会同时存在gcc5 和 gcc7，还需要删除原有gcc链接，重新建立软链接，这样在使用gcc的时候使用的就是gcc5的版本了

• 创建软链接 （gcc 重新升级到gcc7的话，重新创建这里的软链接就可以了）

cd /usr/bin 
sudo rm -r gcc //移除之前的软链接
sudo ln -sf gcc-5 gcc // 建立gcc-5的软链接 此处尝试使用gcc-5.5.0，但是报错
sudo rm -r g++  //同上
sudo ln -sf g++-5 g++  //同上

• 检查gcc 版本

gcc -v  

4. 安装apex

注意，不能直接使用

git clone https://github.com/NVIDIA/apex

因为apex版本与cuda版本不符合，使用旧apex版本，一个旧版本下载连接（我也不知道是什么版本）https://codeload.github.com/NVIDIA/apex/zip/6af09dd938adf11baedabd89da6cd092d3959081，下载完成后解压到项目里。然后执行：

cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./