第一步:从官方网站下载 Miniconda,链接Miniconda — miniconda documentation
第二步:创建一个 conda 环境并激活它
conda create --name openmmlab(可以修改) python=3.8(根据你的python版本) -y
激活环境
conda activate openmmlab
第三步:安装 PyTorch,按照官方说明进行操作,例如
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
此处要参考自己的cuda版本,网上有很多例子
检验是否安装好,命令行输入python
然后运行
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用 TRUE
print(torch.cuda.device_count()) # 查看可用的CUDA数量 1
print(torch.version.cuda) # 查看CUDA的版本号 版本
结果为后面的证明安装成功

本文详细指导如何在Linux系统上通过Miniconda创建Python环境,安装PyTorch及其相关库MMEngine、MMCV、MMDetection和MMPose,以及从源代码安装mmaction2。
最低0.47元/天 解锁文章
254





