总流程:(图片截取自这个网站GET STARTED — MMDetection 3.3.0 documentation)
图片上的链接:https://mmcv.readthedocs.io/en/latest/get_started/installation.html
1. 安装conda
Linux服务器miniconda安装及使用教程_在linux安装mini conda-CSDN博客
列出可用的conda环境
conda env list
删除指定环境
conda remove --name myenv --all
创建并激活指定环境
conda create --name openmmlab python=3.8 -y
conda activate openmmlab
2. 装pytorch,版本别装错了(torch版本和cuda版本都要看)
记住package选conda。
Start Locally | PyTorch用这里面的命令
nvcc --version
看cuda版本(实验室服务器是11.0,可以用这个11.8的 )
(另外顺便一提,nvidia-smi可以看到显卡使用情况,和最高可以用的cuda版本。此处用上面的nvcc --version就行。)
检查torch和cuda版本的命令:
import torch
# 查看 PyTorch 版本
print("PyTorch version:", torch.__version__)
# 检查是否可用 CUDA
print("CUDA available:", torch.cuda.is_available())
命令行敲python,进入解释器,ctrl+d退出
除了cuda版本,还要关注torch版本,我在mmcv的官网(Installation — mmcv 2.1.0 documentation)看到这个
草了,重新下torch,找到torch历史版本。
Previous PyTorch Versions | PyTorch
下这个cuda11.0的。
3. 装别的
GET STARTED — MMDetection 3.3.0 documentation
配好了跑demo就是这样