裸机适配3090,已验证pytorch、cudnn、nvidia驱动版本,避坑指南:
亲测可用版本如下:
系统版本centos7;
显卡:RTX3090;
Nvidia-driver:515;
cuda:11.1;
cudnn:8.2.1;
torch==1.7.1+cu110;
torchvision==0.8.2+cu110;
torchaudio===0.7.2;
pytorch下载命令: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