Linux安装torch1.7.1+torchvision0.8.2+torchaudio0.7.2+mmcv1.4.0

折腾好久，终于把torch安装好了。

1.在.bashrc里面配置cuda的环境变量（已配置好的请忽略）

在终端输入

vim ~/.bashrc

接着按键盘上的  i  进入插入模式

 将上面的配置成你自己的cuda目录和版本。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/extras/CUPTI/lib64 
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.0

按下键盘上的esc退出插入模式，接着输入:wq保存退出

(对linux 的vim等命令不熟悉，网上搜一下)

2.首先创建python环境，选择python3.7（我这里使用anaconda创建环境，推荐大家也用这个）

conda create -n open-mmlab python=3.7 -y

 接着进入这个环境

conda activate open-mmlab

3.因为我的cuda是cuda11.0，所以我到pytorch官网上找到适合我cuda的torch。

我选择的是torch1.7.1+torchvision0.8.2+torchaudio0.7.2，所以利用浏览器的查找功能ctr+f，输入11.0找到cuda11.0对应的torch，选择Wheel下面的这句安装代码，复制到Linux的终端运行。

 

 

在你activate进入到open-mmlab这个环境后，使用命令

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

安装好torch后，查看一下

 到此位置torch就安装好了哦。

4.安装mmcv，直接使用命令

pip install mmcv==1.4.0

 5.打开pycharm，写一个python代码用来查看torch版本

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)

 安装完成。反正装这个快把我弄死了