前三条可能不适合 Ubuntu 20.04 + RTX3060 配置的人【直接看第 8 条即可】

1. pytest+pyblind11安装
2. 将anaconda环境下的 /home/zzy/anaconda3/lib/python3.6/site-packages/torch/utils/cpp_extension.py文件
   将[‘ninja’,’-v’]改成[‘ninja’,’–v’] 或者[‘ninja’,’–version’] 1631/1437 行
3. conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
4. 大佬修改的是配高版本pytorch的源码
   （1）pytorch1.7+cuda11.0-11.2都行，cuda 版本可以尽量的高。
   （2）安装 cuda 切忌不能直接安装 deb 版本，因为会自带驱动并覆盖原来配置，建议岁时悲愤之前环境的安装包。
5. git 切换版本：
   （1）git clone
   （2）查看分支信息：git branch -av
   （3）切换分支：git checkout -b 分支
6. nvrtc兼容问题
   CUDA 11.0 支持 sm_80. 但是 CUDA 11.0中的nvrtc 不支持compute_80.
   RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)
   pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
   参考链接
7. 删除 cudnn：

sudo rm -rf /usr/local/cuda/lib64/libcudnn*
sudo rm -rf /usr/local/cuda/include/cudnn*

8. 适用 pytorch 1.7 的 DCNv2 的 github 链接：链接
9. cuda 路径设置：

if [ $LD_LIBRARY_PATH ]; then
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
else
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64
fi

if [ $PATH ]; then
    export PATH=$PATH:/usr/local/cuda/bin
else
    export PATH=/usr/local/cuda/bin
fi

if [ $CUDA_HOME ]; then
    export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
else
    export CUDA_HOME=/usr/local/cuda
fi