背景

在安装openmmlab的mmsegmentation时，需要安装pytorch。
机子显卡为GeForce RTX 3090，安装cuda为11.4
查看显卡型号方法： 1.直接nvidia-smi。 有个问题是，如果型号信息过长，是无法显示完整的
2. lspci | grep -i vga ，得到处理器跟显卡的十六进制数字代码。 然后到http://pci-ids.ucw.cz/mods/PC/10de?action=help?help=pci， 把数字代码输入即可获取显卡型号。
在尝试pytorch1.8跟1.9的版本都失败了。

import torch
torch.cuda.is_available()
# 输出为True，则代表可调用，Fasle则失败

解决

pytorch安装版本为1.10或者1.11版本

pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

注意

安装完pytorch后，需要重新安装mmcv-full，跟mmsgementation