这俩都差不多。
一、安装配置anaconda:
如何在Linux服务器上安装Anaconda(超详细)_linux安装anconda-CSDN博客
省流:
1. 在官网找到想要的anaconda版本
https://repo.anaconda.com/archive/
wget https://repo.anaconda.com/archive/Anaconda3-2023.03-0-Linux-x86_64.sh
2. 安装
# 设置可执行
chmod +x Anaconda3-5.3.0-Linux-x86_64.sh
# 安装
bash Anaconda3-5.3.0-Linux-x86_64.sh
运行时敲回车,输入yes,否则可能需要自己配置环境变量。
3. 检查
conda -V
输出版本号。
二、创建虚拟环境:
# 新建虚拟环境
conda create -n torch112_py310 python=3.10
# 激活环境
conda activate torch112_py310
三. 安装配置torch:
1. 查看CUDA版本
# 查看cuda版本
nvidia-smi
2. 到官网找适合版本的pytorch
我用的服务器版本是11.4,选择torch1.12版本,注意要在激活的虚拟环境下载。
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
-c 表示channel,外网可能速度慢。可以搜索国内镜像换源试试。
3. 测试安装成功
1)首先输入: python 进入python交互式运行的模式。
2)然后在输入:import torch 不报错即成功。
3)注意可以conda list查看装的torch版本,服务器有显卡的话不要装成cpu的。
四、安装配置mmdet
1. 下载mmdet项目
地址:GitHub - open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark
git clone https://github.com/open-mmlab/mmdetection.git
2. 下载依赖
pip install -U openmim -i https://pypi.tuna.tsinghua.edu.cn/simple
mim install mmengine -i https://pypi.tuna.tsinghua.edu.cn/simpl
mim install "mmcv>=2.0.0" -i https://pypi.tuna.tsinghua.edu.cn/simple
3. 安装mmdet
# 进入项目目录
cd mmdetection
# setup 修改了源码的时候也要用这个命令
python setup.py install