本人用的服务器版本是cuda11.4(实际安装的时候按照11.3版本)
1.使用 conda 新建虚拟环境,并进入该虚拟环境
conda create -n openmmlab python=3.7 -y
conda activate openmmlab
这里的openmmlab是随便起的名字,当然你也可以随便叫
2.基于 PyTorch 官网安装 PyTorch 和 torchvision
conda install pytorch torchvision -c pytorch
这里我的服务器是cuda11.4,我找的是最接近的11.3进行适配的,因此我torch安装的版本是1.10.1
# CUDA 11.3
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge

同时我也建议根据上面的表格来找,然后去pytorch官网找就行
历史版本的记录在这里:Previous PyTorch V

本文介绍在Python环境下安装相关库的步骤。先使用conda新建虚拟环境,基于PyTorch官网安装对应cuda版本的PyTorch和torchvision,再用conda下载mmcv,需指定兼容版本,最后下载mmdet,要注意mmcv和mmdet的对应关系,安装完成后可进行测试。
最低0.47元/天 解锁文章
3989





