折腾好久，终于把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)
安装完成。反正装这个快把我弄死了
Original: https://blog.csdn.net/a_cherry_blossoms/article/details/123669131
Author: LeiHanhan
Title: Linux安装torch1.7.1+torchvision0.8.2+torchaudio0.7.2+mmcv1.4.0
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/618925/
转载文章受原作者版权保护。转载请注明原作者出处！