faster rcnn 环境配置——pytorch0.4

faster rcnn 环境配置

一些坑！

1.装0.4.1版本的torch！！！

以下是0.4.0会碰见的bug
ModuleNotFoundError: No module named ‘torch.autograd’
‘ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.’

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

conda install pytorch=0.4.1 cuda90 -c pytorch

2. 安准cuda9.0，gcc g++ 版本降到5.5

3. cuda9.0对应torchvision0.2.2

4. make.sh 设置CUDA_ARCH

链接：

1.在Ubuntu 18.04中配置GPU环境：安装CUDA 9.0等

https://blog.csdn.net/baimafujinji/article/details/89506896

2.多版本cuda安装：

https://blog.csdn.net/qq_34638161/article/details/80845366
多版本安装时，复制cudnn要复制到对应的cuda版本文件夹，不要放在cuda文件夹中。cuda文件夹只是一个软连接，根据cuda版本的切换而进行删除