python 目标检测yoloV5搭建pytorch环境、paddle环境

pytorch环境

创建环境

conda create -n pytorch python=3.8

执行如下命令，激活这个环境。conda activate 虚拟环境名称

conda activate pytorch

Pytorch环境安装

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

测试安装完成

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

paddle环境

创建环境

conda create -n paddle python=3.8

执行如下命令，激活这个环境。conda activate 虚拟环境名称

conda activate paddle 

paddlepaddle环境安装

conda install paddlepaddle-gpu==2.6.1 cudatoolkit=11.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge

测试安装完成

import paddle
print(paddle.utils.run_check())